Developers
Download
[object Object][object Object][object Object]undefined
[object Object]
  • 算子功能:实现兼容tf.compat.v1.scatter_mul的功能,将tensor updates中的值按指定的索引tensor indices逐切片乘到tensor varRef的切片上。若有多于一个updates值作用到varRef的同一个切片,那么这些值将会在这一切片上连乘。规则如下:
varRef[indices[i],...]=varRef[indices[i],...]×updates[i,...]varRef[indices[i],...] = varRef[indices[i],...] \times updates[i,...] [object Object]

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

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

    [object Object]
  • 返回值

    返回aclnnStatus状态码,具体参见

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

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

    [object Object]
  • 返回值

    返回aclnnStatus状态码,具体参见

[object Object]
  • varRef与updates的数据类型需一致。
  • 需满足shape约束:updates.shape = indices.shape + varRef.shape[1:]。
  • 确定性计算:
    • aclnnScatterMul默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
[object Object]

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

[object Object]