开发者
下载
[object Object][object Object][object Object]undefined
[object Object]
  • 算子功能:实现兼容tf.compat.v1.scatter_min的功能,将tensor updates中的值按指定的索引tensor indices逐元素取最小值更新到tensor varRef的切片上。若有多于一个updates值作用到varRef的同一个切片,那么依次在这一切片上取最小值。规则如下:
varRef[indices[i],...]=min(varRef[indices[i],...],updates[i,...])varRef[indices[i],...] = \min(varRef[indices[i],...], updates[i,...]) [object Object]

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

[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:]。
  • 确定性计算:
    • aclnnScatterMin默认确定性实现。
[object Object]

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

[object Object]