已关闭
[Bug-Report|缺陷反馈]: ops-nn/index/scatter_elements的README里写的不是ScatterElements算子的功能 #1954
Ding_Jing创建于  4月7日关闭于  5月22日
Ding_Jing
Ding_Jing成员
4月7日 创建

Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.

Describe the current behavior / 问题描述 (Mandatory / 必填)

ops-nn/index/scatter_elements的README里写的不是ScatterElements算子的功能,放的是unique。
https://gitcode.com/cann/ops-nn/blob/master/index/scatter_elements/README.md

Environment / 环境信息 (Mandatory / 必填)

全形态

Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)

https://gitcode.com/cann/ops-nn/blob/master/index/scatter_elements/README.md

Describe the expected behavior / 预期结果 (Mandatory / 必填)

预期scatter_elements下放的应该是ScatterElements算子相关的内容

ScatterElements

功能说明

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

参数说明

  • 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算子。

Special notes for this issue/备注 (Optional / 选填)

likedislike
jiangjiawei成员
4月8日 评论:

/assign @sunchun

likedislike
CANN-robot
CANN-robot成员
4月8日 评论:

Notice

This issue can not be assigned to sunchun. Please try to assign to the repository members.

likedislike
Cchenqi317成员
4月13日 将 sunchun 设为负责人
CANN-robotCANN-robot成员
5月22日 关闭了 issue
CANN-robotCANN-robot成员
5月22日 添加了label:resolved