文件最后提交记录最后更新时间
1 个月前
3 个月前
1 个月前
1 个月前
3 个月前
1 个月前
README

ScatterNdMax

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品 ×
Atlas A2 训练系列产品/Atlas A2 推理系列产品 ×
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品 ×
Atlas 训练系列产品 ×

功能说明

  • 算子功能:根据indices在给定变量内,在updates和单个值或切片之间求最大值。
  • 计算公式:ref[indices[k]]=max(ref[indices[k]], updates[k])
    • ref是一个维度为P的张量Tensor;
    • indices是一个维度为Q的整型张量Tensor;
    • indices的shape一定是[d0,..., d_Q-2, K],此处0<K<=P;
    • updates是一个维度为Q−1+P−K的张量:[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]]

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
var 输入 表示一个待被更新的张量,等同于公式中的`ref` DT_INT8、DT_INT16、DT_INT64、DT_UINT64、DT_BOOL、DT_FLOAT16、DT_BF16、DT_FLOAT、DT_INT32、DT_UINT32 ND
indices 输入 一个索引张量,索引到公式中的`ref` INT32、INT64。 ND
updates 输入 使用此张量来更新var张量,必须跟输入var张量保持一样的数据类型 DT_INT8、DT_INT16、DT_INT64、DT_UINT64、DT_BOOL、DT_FLOAT16、DT_BF16、DT_FLOAT、DT_INT32、DT_UINT32 ND
var 输出 表示更新后的张量 DT_INT8、DT_INT16、DT_INT64、DT_UINT64、DT_BOOL、DT_FLOAT16、DT_BF16、DT_FLOAT、DT_INT32、DT_UINT32 ND
use_locking 可选属性 可选属性,默认值为`false`,如果为`true`,这次操作将会被一个lock保护 BOOL -

约束说明

  • 输入shape限制:
    • indices至少是1维,其最后1维的大小不能超过varRef的维度大小。
    • 假设indices最后1维的大小是a,则updates的shape等于indices除最后1维外的shape加上varRef除前a维外的shape。举例:varRef的shape是(4, 5, 6),indices的shape是(3, 2),则updates的shape必须是(3, 6)。

调用说明

调用方式 样例代码 说明
图模式 test_geir_scatter_nd_max 通过GE IR构图方式调用ScatterNdMax算子。