- 算子功能:实现兼容tf.compat.v1.scatter_add和tf.compat.v1.scatter_nd_add的功能,将tensor updates中的值按指定的索引tensor indices加到tensor varRef的切片上。若有多于一个updates值被填入到varRef的同一个切片,那么这些值将会在这一切片上进行累加。规则如下:
或者
每个算子分为,必须先调用“aclnnTfScatterAddGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnTfScatterAdd”接口执行计算。
[object Object]
[object Object]
- 确定性计算:
- aclnnTfScatterAdd默认为非确定性实现,可通过确定性计算配置为确定性实现。
- 需满足以下约束之一:
- updates.shape = indices.shape + varRef.shape[1:]
- indices.shape[-1] <= varRef.shape.rank 且 updates.shape = indices.shape[:-1] + varRef.shape[indices.shape[-1]:]
[object Object]