Star1295
1409
代码
代码
Issues104
Pull Requests210
Actions
项目讨论
Wiki
分析
Star1295
1409
已合并
abs等算子A5实现 #495
ly_cann_coder创建于 2025年12月25日
abs等算子A5实现 #495
已合并
ly_cann_coder创建于 2025年12月25日
从ly_cann_coder/ops-mathdevelop合入到cann/ops-mathmaster
L
ly_cann_coder成员
2025年12月25日

描述

新增以下算子的A5实现

序号 算子 功能
1 abs 为输入张量的每一个元素取绝对值
2 add 对输入完成相加操作
3 addcdiv 张量运算函数，用于执行乘除加组合操作，将张量除法（带缩放）+ 张量加法合并为单个操作
4 addcmul 张量运算函数，用于执行乘除加组合操作，将张量乘法（带缩放）+ 张量加法合并为单个操作
5 axpy 源操作数中每个元素与标量求积后和目的操作数中的对应元素相加
6 bincount 计算非负整数数组中每个数的频率
7 ceil 返回输入tensor中每个元素向上取整的结果
8 clip_by_value_v2 将输入的所有元素限制在一定范围内
9 concat 用于沿指定维度将多个输入 Tensor 进行拼接，输出包含所有输入数据按顺序拼接后的 Tensor
10 concat_v2 用于沿指定维度将多个输入 Tensor 进行拼接，输出包含所有输入数据按顺序拼接后的 Tensor
11 equal 计算两个Tensor是否有相同的大小和元素，返回一个Bool类型
12 exp 返回一个新的张量，该张量的每个元素都是输入张量对应元素的指数
13 fill 对张量进行填充操作，支持非连续的Tensor操作
14 floor 返回输入Tensor中每个元素向下取整，并将结果回填到输入Tensor中
15 floor_div 完成除法计算，对余数向下取整
16 floor_mod 用于计算两个数的模
17 is_neg_inf 判断输入张量的元素是否为负无穷
18 is_pos_inf 判断输入张量的元素是否为正无穷
19 less 判断输入self中的每个元素是否小于输入other的值，返回一个Bool类型的Tensor
20 less_equal 判断输入self中的元素值是否小于等于other的值，并将self的每个元素的值与other值的比较结果写入out中
21 logical_and 对两个输入张量的对应元素执行「与逻辑」判断，输出布尔型张量（True/False）
22 log 对输入张量x的元素，逐元素进行对数计算，并将结果保存到输出张量y中
23 not_equal 逐元素比较两个输入张量是否不相等
24 ones_like 返回形状和类型相同的张量，所有元素都设置为1
25 sign 对输入的tensor逐元素进行Sign符号函数的运算并输出结果tensor
26 sqrt 完成非负数平方根计算，负数情况返回nan
27 sqrt_grad 完成梯度平方根计算
28 square 为输入张量的每一个元素计算平方值
29 squared_difference 为第一个输入张量减去第二个输入张量，并计算其平方值
30 tanh 激活函数，返回与输入tensor shape相同的tensor，对输入tensor进行elementwise的计算
31 tril 将输入的self张量的最后二维（按shape从左向右数）沿对角线的右上部分置零
32 triu 将输入的self张量的最后二维（按shape从左向右数）沿对角线的左下部分置零
33 zeros_like 将张量填充为全零

关联的Issue

测试

文档更新

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
（感谢 ly_cann_coder 的贡献）
CANN-robot
CANN-robot成员
2025年12月25日 评论：

Thank your for your pull-request.

The full list of commands accepted by me can be found at here.

You can get sig-info at here

likedislike
CANN-robot
CANN-robot成员
2025年12月25日 评论：

以下是根据您提交的修改文件推荐的Reviewer和Committer序列，需各模块评审通过后方可合入

Module List Reviewers Committers
repo-cann/ops-math N/A zhou-qilong, gubaocheng, loov1, wangrui_, songkai111
conversion/as_strided N/A loov1, wangrui_, songkai111, FelixTang7, rxtfeng
conversion/fill N/A songkai111, FelixTang7, rxtfeng, zhangyuxiang0119, zhou-qilong
math/abs N/A su-yueming, loov1, wangrui_, zhangyuxiang0119, rxtfeng
math/addcdiv N/A songkai111, FelixTang7, rxtfeng, zhangyuxiang0119, zhou-qilong
conversion/clip_by_value_v2 N/A zhangyuxiang0119, zhou-qilong, gubaocheng, loov1, wangrui_
conversion/concat N/A loov1, songkai111, FelixTang7, rxtfeng, zhangyuxiang0119
conversion/masked_fill N/A zhou-qilong, gubaocheng, loov1, songkai111, FelixTang7
math/add N/A zhangyuxiang0119, zhou-qilong, gubaocheng, loov1, wangrui_
likedislike
CANN-robotCANN-robot成员
2025年12月25日 添加了label：cann-cla/yes
CANN-robot
CANN-robot成员
2025年12月25日 评论：

🔵 source code change are detected, tasks labels is removed in this pull request!

likedislike
L
ly_cann_coder成员
2025年12月25日 评论：

compile

likedislike
此处折叠了215条消息 查看更多
Lly_cann_coder成员
2025年12月27日 修改了pull request 的描述
Lly_cann_coder成员
2025年12月27日 修改了pull request 的描述
RRuiWang_成员
2025年12月27日 解决了最后一个问题
CANN-robot
CANN-robot成员
2025年12月27日 评论：

Review Guide

This Pull-Request Passes Review.
Committers who writed a comment of /approve are: songkai111.
Reviewers who writed a comment of /lgtm are: songkai111.

likedislike
CANN-robotCANN-robot成员
2025年12月27日 合入了pull request