文件最后提交记录最后更新时间
5 个月前
9 个月前
5 个月前
7 个月前
6 个月前
9 个月前
5 个月前
README

ScatterList简介

产品支持情况

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

功能说明

  • 算子功能:将稀疏矩阵更新应用到变量引用中。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
var 输入 表示待被更新的张量,Device侧的aclTensor。 DT_INT8、DT_INT16、DT_INT32、DT_UINT8、DT_UINT16、DT_UINT32、DT_FLOAT16、DT_BF16、DT_FLOAT ND
indice 输入 表示待更新的索引张量,Device侧的aclTensor。 INT32、INT64。 ND
updates 输入 表示需要更新到var上的张量,Device侧的aclTensor。 DT_INT8、DT_INT16、DT_INT32、DT_UINT8、DT_UINT16、DT_UINT32、DT_FLOAT16、DT_BF16、DT_FLOAT ND
mask 输入 表示需要更新数据的掩码,Device侧的aclTensor。 DT_UINT8 ND
var 输出 表示更新后的张量,Device侧的aclTensor。 FLOAT32、FLOAT16、DOUBLE、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、BOOL ND
reduce 属性 HOST侧的字符串,选择应用的reduction操作 string -
axis 属性 用来scatter的维度。 int64_t -
  • Kirin X90/Kirin 9030处理器系列产品:不支持BFLOAT16。

约束说明

无