news 2026/5/22 23:59:39

CANN/pypto copysign函数API文档

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
CANN/pypto copysign函数API文档

# pypto.copysign

【免费下载链接】pyptoPyPTO(发音: pai p-t-o):Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto

产品支持情况

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

功能说明

逐元素地取input的数值和other的符号。计算公式如下:

$$ res_i = input_i * sign(other_i) $$

函数原型

copysign(input: Tensor, other: Tensor) -> Tensor

参数说明

参数名输入/输出说明
input输入源操作数。
支持的类型为Tensor类型。
Tensor支持的数据类型为:DT_FP32, DT_FP16, DT_BF16, DT_INT32, DT_INT16。
不支持空Tensor;Shape仅支持1-4维,支持按照单个维度广播到相同形状;Shape Size不大于2147483647(即INT32_MAX)。
other输入源操作数。
支持的类型为Tensor类型。
Tensor支持的数据类型为:DT_FP32, DT_FP16, DT_BF16, DT_INT32, DT_INT16。
不支持空Tensor;Shape仅支持1-4维,支持按照单个维度广播到相同形状;Shape Size不大于2147483647(即INT32_MAX)。

返回值说明

返回输出Tensor,Tensor的数据类型和input、other相同,Shape为input和other广播后大小。

约束说明

  1. input 和 other 类型应该相同。

调用示例

TileShape设置示例

说明:调用该operation接口前,应通过set_vec_tile_shapes设置TileShape。

TileShape维度应和输出一致。

如非广播场景,输入input shape为[m, n],other为[m, n],输出为[m, n],TileShape设置为[m1, n1], 则m1, n1分别用于切分m, n轴。

广播场景,输入input shape为[m, n],other为[m, 1],输出为[m, n],TileShape设置为[m1, n1], 则m1, n1分别用于切分m, n轴。

pypto.set_vec_tile_shapes(4, 16)

接口调用示例

a = pypto.tensor([3, 3], pypto.DT_FP32) b = pypto.tensor([3, 3], pypto.DT_FP32) out = pypto.copysign(a, b)

结果示例如下:

Input x : [[1 -2 3], [4 5 -6], [-7 8 9]] Input y : [[-1 6 -8], [1 -1 0], [7 -8 9]] Output out:[[-1 2 -3], [4 -5 6], [7 -8 9]]

【免费下载链接】pyptoPyPTO(发音: pai p-t-o):Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/5/22 23:52:30

【软考网络工程师-案例分析易错题整理(下)】

软考网络工程师-案例分析易错题整理(下)一、问题一二、问题二三、问题三四、问题四五、问题五一、问题一 1.某小区采用HFC接入Internet的解决方案进行网络设计,网络结构如图1-1 所示。 【问题1】网络设计流程通常由以下五阶段组成&#xff1…

作者头像 李华
网站建设 2026/5/22 23:48:52

CANN/Ascend C HelloWorld样例

HelloWorld样例 【免费下载链接】asc-devkit 本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。 项目地址: https://gitcode…

作者头像 李华
网站建设 2026/5/22 23:47:16

react-tween-state源码解析:理解React动画库的内部工作原理

react-tween-state源码解析:理解React动画库的内部工作原理 【免费下载链接】react-tween-state React animation. 项目地址: https://gitcode.com/gh_mirrors/re/react-tween-state react-tween-state是一个轻量级的React动画库,它通过状态缓动&…

作者头像 李华
网站建设 2026/5/22 23:37:03

【Linux】基本指令3

【Linux】基本指令1 【Linux】基本指令2 📌 相关专栏 【Linux专栏】【C语言专栏】【测试专栏】 文章目录cal指令find指令which指令whereis指令alias指令grep指令zip/unzip指令tar指令bc指令uname指令cal指令 cal指令可以用来显示公历(阳历)…

作者头像 李华
网站建设 2026/5/22 23:36:03

Java单元测试覆盖率≠质量保障!Gemini动态路径分析揭示87%高危空指针未被UT覆盖(附可落地的增强策略模板)

更多请点击: https://intelliparadigm.com 第一章:Java单元测试覆盖率的认知误区与质量保障本质 许多团队将“80%行覆盖率”奉为质量金标准,却在上线后遭遇隐蔽的空指针或边界逻辑崩溃。覆盖率数字本身不等于质量,它只是测试活动…

作者头像 李华