在指定维度上,根据给定的索引,将源张量中的值加到输入张量中对应位置的值上。
每个算子分为,必须先调用"aclnnIndexAddV2GetWorkspaceSize"接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用"aclnnIndexAddV2"接口执行计算。
[object Object]
[object Object]
- 确定性计算:
- aclnnIndexAddV2默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
- index取值范围
- index输入取值范围在[0,self.shape[dim])范围内,即索引值输入范围为self在dim维度上的shape大小,不支持负索引和越界索引。
- 高性能模式(mode=0)额外限制:
- dim取值为0或-2。
- self数据类型为FLOAT、FLOAT16、INT32、INT16或BFLOAT16。
- self是二维tensor。
- 如果mode=0,且满足参数说明中的限制,但不满足以上三条约束,接口会抛出错误。
[object Object]