开发者
资源
[object Object][object Object][object Object]undefined
[object Object]
  • 算子功能:实现兼容tf.compat.v1.scatter_add和tf.compat.v1.scatter_nd_add的功能,将tensor updates中的值按指定的索引tensor indices加到tensor varRef的切片上。若有多于一个updates值被填入到varRef的同一个切片,那么这些值将会在这一切片上进行累加。规则如下:
varRef[indices[i,...,j],...]=varRef[indices[i,...,j],...]+updatesvarRef[indices[i,...,j],...] = varRef[indices[i,...,j],...] + updates

或者

varRef[indices[i,:]]=varRef[indices[i,:]]+updates[i,...]varRef[indices[i,:]] = varRef[indices[i,:]] + updates[i,...] [object Object]

每个算子分为,必须先调用“aclnnTfScatterAddGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnTfScatterAdd”接口执行计算。

[object Object]
[object Object]
[object Object]
  • 参数说明

    [object Object]
  • 返回值

    返回aclnnStatus状态码,具体参见

    第一段接口完成入参校验,若出现以下错误码,则对应原因为:

    [object Object]
[object Object]
  • 参数说明

    [object Object]
  • 返回值

    返回aclnnStatus状态码,具体参见

[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]

示例代码如下,仅供参考,具体编译和执行过程请参考

[object Object]