ScatterMax

产品支持情况

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

功能说明

• 算子功能：实现兼容tf.compat.v1.scatter_max的功能，将tensor updates中的值按指定的索引tensor indices逐元素取最大值更新到tensor var的切片上。若有多于一个updates值作用到var的同一个切片，则依次在该切片上取最大值。属于原地（in-place）更新，输出复用输入var。规则如下：

  $$var[indices[i],...]=\max (var[indices[i],...],updates[i,...])$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
var	输入	支持空Tensor。表示待被更新的张量，Device侧的aclTensor，原地更新。shape支持1维及以上，数据类型需与updates一致。	FLOAT16、FLOAT32、INT32、INT8、UINT8	ND
indices	输入	支持空Tensor。表示待更新的索引张量，Device侧的aclTensor。shape支持1维及以上，数据类型为INT32或INT64。支持索引越界，越界索引（小于0或大于等于var第一维）会被静默跳过，不做更新。	INT32、INT64	ND
updates	输入	支持空Tensor。表示需要与var逐元素取最大值的张量，Device侧的aclTensor。shape支持1维及以上，需满足updates.shape = indices.shape + var.shape[1:]，数据类型需与var一致。	FLOAT16、FLOAT32、INT32、INT8、UINT8	ND
var	输出	支持空Tensor。表示更新后的张量，Device侧的aclTensor，与输入var共享内存。shape和数据类型均与输入var一致。	FLOAT16、FLOAT32、INT32、INT8、UINT8	ND
use_locking	属性	HOST侧的bool值，是否对更新加锁，默认false。	bool	-

• Ascend 950PR&950DT系列产品：var、updates的数据类型支持FLOAT16、FLOAT32、INT32、INT8、UINT8。
• Atlas A3系列产品、Atlas A2系列产品：var、updates的数据类型支持FLOAT16、FLOAT32、INT32、INT64、DOUBLE，不支持INT8、UINT8。

约束说明

• 确定性计算：默认确定性实现。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_scatter_max	通过aclnnScatterMax接口方式调用ScatterMax算子。