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

ScatterElements

产品支持情况

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

功能说明

算子功能:返回输入张量中的唯一元素。

参数说明

  • self(aclTensor*,计算输入):Device侧的aclTensor,数据格式支持ND,维度不大于8。

    • Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品、Ascend 950PR/Ascend 950DT:数据类型支持BOOL、FLOAT、FLOAT16、DOUBLE、UINT8、INT8、UINT16、INT16、INT32、UINT32、UINT64、INT64、BFLOAT16。
  • sorted(bool,计算输入): 表示是否对 valueOut 按升序进行排序。true时排序,false时乱序。

  • returnInverse(bool,计算输入): 表示是否返回输入数据中各个元素在 valueOut 中的下标。

  • valueOut(aclTensor*,计算输出): Device侧的aclTensor,第一个输出张量,输入张量中的唯一元素,数据格式支持ND。

    • Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品、Ascend 950PR/Ascend 950DT:数据类型支持BOOL、FLOAT、FLOAT16、DOUBLE、UINT8、INT8、UINT16、INT16、INT32、UINT32、UINT64、INT64、BFLOAT16。
  • inverseOut(aclTensor*,计算输出): 第二个输出张量,当returnInverse为True时有意义,返回self中各元素在valueOut中出现的位置下标,数据类型支持INT64,shape与self保持一致。

约束说明

无

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_unique 通过aclnnUnique接口方式调用ScatterElements算子。