pypto.scatter#

产品支持情况#

  • Ascend 950PR/Ascend 950DT：支持

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

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

函数原型#

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_的调用示例。