ScatterNdUpdate

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

• 算子功能：根据indices在给定变量内，将updates应用于变量的单个值或切片。
• 计算公式：ref[indices[k]]=updates[k]
  • ref是一个维度为P的张量Tensor；
  • indices是一个维度为Q的整型张量Tensor；
  • indices的shape一定是[d0,..., d_Q-2, K]，此处0<K<=P；
  • updates是一个维度为Q−1+P−K的张量：[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]]

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
var	输入	表示一个待被更新的张量,等同于公式中的`ref`	DT_INT8、DT_INT64、DT_BOOL、DT_FLOAT16、DT_BF16、DT_FLOAT、FLOAT8_E4M3FN、FLOAT8_E5M2、HIFLOAT8	ND
indices	输入	一个索引张量，索引到公式中的`ref`	INT32、INT64。	ND
updates	输入	使用此张量来更新var张量，必须跟输入var张量保持一样的数据类型	DT_INT8、DT_INT64、DT_BOOL、DT_FLOAT16、DT_BF16、DT_FLOAT、FLOAT8_E4M3FN、FLOAT8_E5M2、HIFLOAT8	ND
var	输出	表示更新后的张量	DT_INT8、DT_INT64、DT_BOOL、DT_FLOAT16、DT_BF16、DT_FLOAT、FLOAT8_E4M3FN、FLOAT8_E5M2、HIFLOAT8	ND
use_locking	可选属性	可选属性，默认值为`false`,如果为`true`,这次操作将会被一个lock保护	BOOL	-

Atlas A3 训练系列产品/Atlas A3 推理系列产品、Atlas A2 训练系列产品/Atlas A2 推理系列产品：FLOAT8_E4M3FN、FLOAT8_E5M2、HIFLOAT8数据类型在该型号中不支持。

约束说明

• 输入shape限制：
  • indices至少是2维，其最后1维的大小不能超过varRef的维度大小。
  • 假设indices最后1维的大小是a，则updates的shape等于indices除最后1维外的shape加上varRef除前a维外的shape。举例：varRef的shape是(4, 5, 6)，indices的shape是(3, 2)，则updates的shape必须是(3, 6)。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_scatter_nd_update	通过aclnnScatterNdUpdate接口方式调用ScatterNdUpdate算子。