pypto.index_add#

产品支持情况#

  • Ascend 950PR/Ascend 950DT：支持

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

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

功能说明#

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

函数原型#

index_add(input: Tensor, dim: int, index: Tensor, source: Tensor, *, alpha: Union[int, float] = 1) -> Tensor

参数说明#

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

返回值说明#

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

约束说明#

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

调用示例#

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