news 2026/6/24 17:02:04

CANN/pypto amin函数API文档

作者头像

张小明

前端开发工程师

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

# pypto.amin

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

功能说明

对一个多维向量在指定的维度求最小值。

定义指定计算的维度(Reduce轴)为R轴,非指定维度(Normal轴)为A轴。如下图所示,对Shape为(2, 3)的二维矩阵进行运算,指定在第一维求最小值,输出结果为[1, 2, 3];指定在第二维求最小值,输出结果为[1, 4]。

图 1amin按第一个维度计算示例

图 2amin按最后一个维度计算示例

函数原型

amin(input: Tensor, dim: int, keepdim: bool = False) -> Tensor:

参数说明

参数名输入/输出说明
input输入源操作数。
支持的类型为:Tensor。
Tensor支持的数据类型为:DT_FP16,DT_BF16,DT_FP32,DT_INT32,DT_INT16。
不支持空Tensor;Shape仅支持1-4维;Shape Size不大于2147483647(即INT32_MAX)。
dim输入源操作数。
支持任意单轴。
keepdim输入源操作数
控制在进行归约后,是否保持被压缩的维度。
默认值为False。

返回值说明

返回输出Tensor,输出Tensor的Shape与keepdim参数相关。

若keepdim参数为 True,则在执行归约操作后保留被归约的维度。输出Tensor在除dim指定的维度外,其他维度的Shape与输入Tensor的Shape一致,而在dim指定的维度上的大小为 1。

若keepdim参数为 False(默认),则被归约的维度会从输出Tensor中移除,而tileshape中对应的维度不变, 所以建议在调其他operation前重设tileshape。

约束说明

  1. TileShape大小不超过 64KB;

  2. 尾轴要 32bytes 对齐;

调用示例

TileShape设置示例

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

TileShape维度应和输入input一致。

示例1:输入input shape为[m, n],输出为[m, 1],TileShape设置为[m1, n1], 则m1, n1分别用于切分m, n轴。

pypto.set_vec_tile_shapes(4, 16)

注意:如果keepdim设置为false,则被归约的维度会从输出Tensor中移除,而tileshape中对应的维度不变, 所以建议在调其他operation前重设tileshape。

接口调用示例

x = pypto.tensor([2, 3], pypto.DT_FP32) y = pypto.amin(x, -1, True)

结果示例如下:

输入数据 x: [[1.0 2.0 3.0], [1.0 2.0 3.0]] 输出数据 y: [[1.0], [1.0]]

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

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

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

Balena Etcher终极指南:三步安全烧录操作系统镜像的完整教程

Balena Etcher终极指南:三步安全烧录操作系统镜像的完整教程 【免费下载链接】etcher Flash OS images to SD cards & USB drives, safely and easily. 项目地址: https://gitcode.com/GitHub_Trending/et/etcher Balena Etcher是一款简单快速的开源镜像…

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

AI驱动的混合动力公交调度与能耗优化:从理论到工程实践

1. 项目概述与核心价值最近几年,城市公交系统的智能化转型是个热门话题,但真正深入到运营层面,尤其是针对混合动力这种“双动力”车型的调度与能耗优化,能聊透的并不多。我花了近半年时间,围绕“基于AI的混合动力公交车…

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

CANN/pypto设置立方体切片形状

pypto.set_cube_tile_shapes 【免费下载链接】pypto PyPTO(发音: pai p-t-o):Parallel Tensor/Tile Operation编程范式。 项目地址: https://gitcode.com/cann/pypto 产品支持情况 产品是否支持Ascend 950PR/Ascend 950DT√Atlas A2 …

作者头像 李华
网站建设 2026/5/9 23:51:24

CANN/GE运行时执行系统——从OM文件到硅片计算的桥梁

运行时执行系统——从 OM 文件到硅片计算的桥梁 【免费下载链接】ge GE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。…

作者头像 李华
网站建设 2026/5/9 23:49:50

大语言模型越狱提示词原理与实战:从安全机制博弈到提示工程进阶

1. 项目概述:当GPT遇上“越狱”如果你深度使用过ChatGPT、Claude这类大语言模型,一定遇到过这样的时刻:你希望它帮你写一段更具批判性的分析,或者模拟一个它通常“不愿”扮演的角色,但得到的回复往往是“作为一个人工智…

作者头像 李华