已合并
新增ScatterNdMax/ScatterNdMin算子 #3899
z30075199创建于 4月16日
新增ScatterNdMax/ScatterNdMin算子 #3899
已合并
z30075199创建于 4月16日
z30075199成员
4月16日

描述

新增ScatterNdMax/ScatterNdMin算子
scatter_nd_max:根据indices在给定变量内,在updates和单个值或切片之间求最大值

参数名 输入/输出/属性 描述 数据类型 数据格式
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 -

scatter_nd_min:根据indices在给定变量内,在updates和单个值或切片之间求最小值

参数名 输入/输出/属性 描述 数据类型 数据格式
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 -

关联的Issue

https://gitcode.com/cann/ops-nn/issues/2919

测试

测试泛化用例200条,功能精度达标

文档更新

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 z30075199 的贡献)
Zz30075199成员
4月16日 创建了 pull request,commit 6a66e208
CANN-robot
CANN-robot成员
4月16日 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here
You can get sig-info at here


PR Approval Progress

Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
*/*/README.md 陈娇, 杨阳, 刘波 (3/2) 陈娇 (1/1)
*/*/op_graph/*_proto.h 王永光, 刘波, 杨阳 (3/2) 王永光 (1/1)
*/*/op_host/*_def.cpp 王永光, 杨阳, 刘波 (3/2) 王永光 (1/1)
docs 刘波, 杨阳 (2/2) 刘波 (1/1)
index 王永光, 刘波, 杨阳 (3/2) 王永光, 刘波 (2/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

z30075199, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
CANN-robotCANN-robot成员
4月16日 添加了label:cann-cla/yes
CANN-robotCANN-robot成员
4月16日 将chenfeng61,FelixTang7,crystalhu,yangyang016,fanqirui,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为评审人
CANN-robotCANN-robot成员
4月16日 将chenfeng61,FelixTang7,crystalhu,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为审查人
此处折叠了182条消息 查看更多
CANN-robot
CANN-robot成员
5月28日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: 陈娇, 王永光, 刘波.
Reviewers who wrote a comment of /lgtm are: 王永光, 杨阳, 陈娇, 刘波.

likedislike
CANN-robot
CANN-robot成员
5月28日 评论:

The MR can not be merged, because of CodeReview discussion not resolved

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike
Zz30075199成员
5月28日 解决了最后一个问题
CANN-robot
CANN-robot成员
5月28日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: 王永光, 陈娇, 刘波.
Reviewers who wrote a comment of /lgtm are: 陈娇, 刘波, 杨阳, 王永光.

likedislike
CANN-robotCANN-robot成员
5月28日 合入了pull request