- 算子功能:实现兼容tf.compat.v1.scatter_max的功能,将tensor updates中的值按指定的索引tensor indices逐元素取最大值更新到tensor varRef的切片上。若有多于一个updates值作用到varRef的同一个切片,那么依次在这一切片上取最大值。规则如下:
每个算子分为,必须先调用“aclnnScatterMaxGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnScatterMax”接口执行计算。
[object Object]
[object Object]
- varRef与updates的数据类型需一致。
- 需满足shape约束:updates.shape = indices.shape + varRef.shape[1:]。
- 确定性计算:
- aclnnScatterMax默认确定性实现。
[object Object]