news 2026/5/9 16:06:36

CANN/pypto maximum逐元素最大值API

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
CANN/pypto maximum逐元素最大值API

# 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/9 16:04:56

打造你的专属游戏云:用Sunshine开启跨平台串流新体验

打造你的专属游戏云:用Sunshine开启跨平台串流新体验 【免费下载链接】Sunshine Self-hosted game stream host for Moonlight. 项目地址: https://gitcode.com/GitHub_Trending/su/Sunshine 你是否想过,将书房里的高性能游戏PC变成一台24小时在线…

作者头像 李华
网站建设 2026/5/9 16:04:55

Java 设计模式:最佳实践与应用

Java 设计模式:最佳实践与应用 核心概念 设计模式是解决软件设计中常见问题的经典解决方案。掌握设计模式可以提高代码的可复用性、可维护性和可扩展性。本文将介绍常用的设计模式及其在 Java 中的最佳实践。 创建型模式 1. 单例模式 // 饿汉式单例 public class Ea…

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

CANN图像算子双三次上采样反向

aclnnUpsampleBicubic2dBackward 【免费下载链接】ops-cv 本项目是CANN提供的图像处理、目标检测相关的算子库,实现网络在NPU上加速计算。 项目地址: https://gitcode.com/cann/ops-cv 📄 查看源码 产品支持情况 产品是否支持 Ascend 950PR/Asc…

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

别再折腾了!Windows 10/11下EDK2+VS2019+Python 3.11一键式环境搭建保姆级教程

Windows 10/11下EDK2开发环境一键式搭建终极指南 每次打开EDK2的官方文档,看到那密密麻麻的依赖项和复杂的配置步骤,是不是感觉头都大了?作为一名曾经被EDK2环境搭建折磨到怀疑人生的开发者,我完全理解那种面对十几个安装包、无数…

作者头像 李华