news 2026/5/22 17:22:16

CANN/pypto最大元素函数

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
CANN/pypto最大元素函数

# pypto.maximum

【免费下载链接】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 推理系列产品

功能说明

计算输入与另一输入的逐元素最大值。支持二维、三维或四维的Tensor。

注意事项

  • 不支持 SymbolicScalar 参数:如果需要对 SymbolicScalar 进行比较,请使用 SymbolicScalar.max() 方法
  • 两个参数中至少一个为 Tensor 类型

函数原型

maximum( input: Union[Tensor, Element, int, float], other: Union[Tensor, Element, int, float] ) -> Tensor

参数说明

参数名输入/输出说明
input输入源操作数。
支持的类型为 int, float, Element, Tensor类型。
当为 int 或者 float 类型时会自动转换为 Element 类型,其中 int 对应 DT_INT32,float 对应 DT_FP32。当需要使用其他数据类型时,可以通过 Element 构建。
Tensor和Element支持的数据类型为:DT_FP16,DT_BF16, DT_INT16,DT_INT32,DT_FP32
不支持空Tensor;Shape仅支持1-4维;Shape Size不大于2147483647(即INT32_MAX)。
other输入源操作数。
支持的类型为 int, float, Element, Tensor类型。
当为 int 或者 float 类型时会自动转换为 Element 类型,其中 int 对应 DT_INT32,float 对应 DT_FP32。当需要使用其他数据类型时,可以通过 Element 构建。
Tensor和Element支持的数据类型为:DT_FP16,DT_BF16, DT_INT16,DT_INT32,DT_FP32
不支持空Tensor;Shape仅支持1-4维;Shape Size不大于2147483647(即INT32_MAX)。
类型和数据类型必须与源操作数一保持一致。

源操作数一与源操作数二之间至少一者为Tensor。

返回值说明

当两个源操作数均为Tensor时,两个Tensor必须满足广播关系。该接口返回一个与源操作数一和源操作数二广播后形状相同的Tensor,数据类型与源操作数相同,其元素为源操作数一和源操作数二的逐元素最大值。且源操作数为Tensor时,源操作数一和源操作数二均支持多轴广播。

当两个源操作数之中存在一个Tensor时,返回与输入Tensor相同形状的Tensor,其元素为源操作数一和源操作数二的逐元素最大值。

调用示例

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], pypto.DT_INT32) b = pypto.tensor([3], pypto.DT_INT32) out = pypto.maximum(a, b)

结果示例如下:

输入数据a: [0, 2, 4] 输入数据b: [3, 1, 3] 输出数据out: [3, 2, 4]

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

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

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

AI 工作范式下的研发新范式:从需求到测试的全链路落地指南

写在前面 最近一年,团队里几乎每个 Java 后端、前端、甚至产品经理,都在用 AI 编辑器写代码。Cursor、Qoder、Claude Code、Trae、Copilot……工具的迭代速度肉眼可见。 但用着用着大家发现一个尴尬的现实:工具升级了,研发流程没升…

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

PDF Arranger终极指南:快速掌握免费PDF页面管理工具

PDF Arranger终极指南:快速掌握免费PDF页面管理工具 【免费下载链接】pdfarranger Small python-gtk application, which helps the user to merge or split PDF documents and rotate, crop and rearrange their pages using an interactive and intuitive graphic…

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

B站直播弹幕机器人实战手册:3步打造高互动自动化直播间

B站直播弹幕机器人实战手册:3步打造高互动自动化直播间 【免费下载链接】MagicalDanmaku 本仓库及所有相关项目已永久停止开发、维护和任何形式的分发。 项目地址: https://gitcode.com/gh_mirrors/bi/MagicalDanmaku 还在为直播时手忙脚乱回复弹幕而焦虑吗&…

作者头像 李华
网站建设 2026/5/22 17:15:59

Git-Sim技术工具:可视化分析Git操作的终极开发效率解决方案

Git-Sim技术工具:可视化分析Git操作的终极开发效率解决方案 【免费下载链接】git-sim Visually simulate Git operations in your own repos with a single terminal command. 项目地址: https://gitcode.com/gh_mirrors/gi/git-sim 在复杂的软件开发流程中&…

作者头像 李华
网站建设 2026/5/22 17:15:53

NetCoMi微生物网络分析:5步完整安装与快速入门指南

NetCoMi微生物网络分析:5步完整安装与快速入门指南 【免费下载链接】NetCoMi Network construction, analysis, and comparison for microbial compositional data 项目地址: https://gitcode.com/gh_mirrors/ne/NetCoMi NetCoMi是一个专门为微生物组数据分析…

作者头像 李华
网站建设 2026/5/22 17:15:25

免费解锁AMD Ryzen隐藏性能:SMUDebugTool完全指南

免费解锁AMD Ryzen隐藏性能:SMUDebugTool完全指南 【免费下载链接】SMUDebugTool A dedicated tool to help write/read various parameters of Ryzen-based systems, such as manual overclock, SMU, PCI, CPUID, MSR and Power Table. 项目地址: https://gitcod…

作者头像 李华