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

ScatterNd

产品支持情况

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

功能说明

算子功能:拷贝data的数据至out,同时在指定indices处根据updates更新out中的数据。

参数说明

  • data(aclTensor*,计算输入):Device侧的aclTensor,数据类型与updates、out一致,shape满足1<=rank(data)<=8。支持非连续的Tensor数据格式支持ND。

    • Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:数据类型支持FLOAT16、FLOAT、BOOL、BFLOAT16
  • indices(aclTensor*,计算输入):Device侧的aclTensor,数据类型支持INT32、INT64。indices.shape[-1] <= rank(data),且1<=rank(indices)<=8。支持非连续的Tensor数据格式支持ND。仅支持非负索引。indices中的索引数据不支持越界。

  • updates(aclTensor*,计算输入):Device侧的aclTensor,数据类型与data、out一致。shape要求rank(updates)=rank(data)+rank(indices)-indices.shape[-1] -1,且满足1<=rank(updates)<=8。支持非连续的Tensor数据格式支持ND。

    • Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:数据类型支持FLOAT16、FLOAT、BOOL、BFLOAT16
  • out(aclTensor*,计算输出):Device侧的aclTensor,数据类型与data、out一致,shape与data一致。支持非连续的Tensor数据格式支持ND。

    • Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:数据类型支持FLOAT16、FLOAT、BOOL、BFLOAT16

约束说明

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_scatter_nd 通过aclnnScatterNd接口方式调用ScatterNdUpdate算子。