CANN逆排列算子文档

张开发
2026/5/10 2:53:49 15 分钟阅读

分享文章

CANN逆排列算子文档
InvertPermutation【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品×Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能计算输入张量x的逆排列。计算公式y[x[i]] i 其中 0 i len(x)参数说明参数名输入/输出/属性描述数据类型数据格式x输入输入张量。INT32, INT641Dy输出输入张量x的逆排列INT32, INT641D约束说明无。调用说明调用方式样例代码说明图模式调用test_geir_invert_permutation通过算子IR构图方式调用invert_permutation算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

更多文章