pypto.index_add_ub#

产品支持情况#

  • Ascend 950PR/Ascend 950DT：支持

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

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

功能说明#

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

函数原型#

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

参数说明#

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

返回值说明#

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

约束说明#

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

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

调用示例#

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