pypto.scatter#

产品支持情况#

  • Ascend 950PR/Ascend 950DT：支持

  • Atlas A3 训练系列产品/Atlas A3 推理系列产品：支持

  • Atlas A2 训练系列产品/Atlas A2 推理系列产品：支持

功能说明#

将src的值写入input中。写入位置由index指定。3维计算公式如下，其他维度以此类推：

src为固定标量时： $\( \begin{cases} input\left[ index\left[i\right]\left[j\right]\left[k\right] \right]\left[j\right]\left[k\right] = src & \text{if } dim = 0 \\ input\left[i\right]\left[ index\left[i\right]\left[j\right]\left[k\right] \right]\left[k\right] = src & \text{if } dim = 1 \\ input\left[i\right]\left[j\right]\left[ index\left[i\right]\left[j\right]\left[k\right] \right] = src & \text{if } dim = 2 \end{cases} \)$

src为Tensor时： $\( \begin{cases} input\left[ index\left[i\right]\left[j\right]\left[k\right] \right]\left[j\right]\left[k\right] = src\left[i\right]\left[j\right]\left[k\right] & \text{if } dim = 0 \\ input\left[i\right]\left[ index\left[i\right]\left[j\right]\left[k\right] \right]\left[k\right] = src\left[i\right]\left[j\right]\left[k\right] & \text{if } dim = 1 \\ input\left[i\right]\left[j\right]\left[ index\left[i\right]\left[j\right]\left[k\right] \right] = src\left[i\right]\left[j\right]\left[k\right] & \text{if } dim = 2 \end{cases} \)$

函数原型#

scatter(input: Tensor, dim: int, index: Tensor, src: Union[float, Element, Tensor], *, reduce: str = None) -> Tensor

scatter_的non-inplace版本，可参考pypto.scatter_

参数说明#

请参考pypto.scatter_的参数说明。

返回值说明#

请参考pypto.scatter_的返回值说明。

约束说明#

  1. 请参考pypto.scatter_的约束说明。

  2. Tensor类型输入不支持TileOpFormat.TILEOP_NZ格式。

调用示例#

请参考pypto.scatter_的调用示例。