randperm算子不支持导出onnx

最新推荐文章于 2024-02-21 18:06:14 发布
最新推荐文章于 2024-02-21 18:06:14 发布
阅读量1.1k 收藏
点赞数
CC 4.0 BY-SA版权
分类专栏： pytorch ONNX 深度学习算法部署 文章标签： onnx pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/yangjinyi1314/article/details/124312908
当尝试将PyTorch的torch.randperm算子导出为ONNX格式时，会发现该操作不被支持。一种解决方法是改用random.shuffle对Tensor进行洗牌操作。参考相关博客和论坛讨论，可以找到将torch.randperm功能替换的替代方案。
问：想跟您请教下torch.randperm这个算子转onnx不支持应该怎么实现？
答复：可以先用 ONNX_FALLTHROUGH转成onnx，然后再给TensorRT写个plugin，转出来后也可以用ONNX-graphsurgeon把这个node展开成TRT支持的node（如果可以等效替换掉的话）。
注意：好像没法用TRT layer表示randperm：torch.randperm — PyTorch 1.11.0 documentation

有人也给了其他思路，比如：用random.shuffle 改写下！

参考：

PyTorch | torch.randperm()使用方法_SophiaCV的博客-CSDN博客_pytorch中randperm

了解本专栏 订阅专栏 解锁全文 超级会员免费看
67、ncnn学习onnx2ncnn不支持五维算子reshape/transpose转换方法学习
sxj731533730
07-11 537
一点点剖析ncnn的不支持，这里先不考虑batch>1的情况 输入维度是 1x3x6x6 使用reshape转成1x6x2x3x3 对应ncnn的内存布局是cdhw 所以修改param结果为。一、例子一，先造个五维度的reshape转换，ncnn的onnx2ncnn不支持转换，参考ppocr_4算子。基本思想：学习ncnn转模型集锦，仅在onnx2ncnn使用， 先不谈pnnx。执行生成pt使用onnx转一下，我用mingw+window转一下。测试结果，ncnn不支持五维度输出，直接崩溃。
常见ONNX算子节点
欧吼的CSDN
03-14 4144
ONNX（Open Neural Network Exchange）是一个开放格式，用于表示深度学习模型，允许模型被不同的深度学习框架所使用。ONNX定义了一系列的算子（Operators），每个算子对应一个特定的操作，例如卷积、池化、归一化等。请注意，ONNX算子集会随着ONNX版本的更新而增加新的算子或修改现有算子。DepthToSpace - 深度到空间的转换。SpaceToDepth - 空间到深度的转换。ArgMax - 沿指定轴的元素最大值索引。ArgMin - 沿指定轴的元素最小值索引。
pytorch导出onnx时遇到不支持的算子怎么解决
zhaoyqcsdn的博客
01-10 3446
然后再自己实现custom_domain::customOp2这个算子，如果用TensorRT，就需要自己实现一个插件。
yolov5导出onnx模型报错：Unsupported ONNX opset version: 17
李昂的博客
05-02 2356
2、onnx算子文档，https://github.com/onnx/onnx/blob/main/docs/Operators.md。Win10系统下使用pytorch1.11.0、onnx1.12.0转换yolov5cls转换onnx报错如下。1、模型部署，https://zhuanlan.zhihu.com/p/498425043。查找了相关资料后是由于pytorch对onnx算子的不支持。
模型转换案例学习：等效替换不支持算子
csdnsqst0050的博客
02-21 1978
高通SNPE支持Caffe、ONNX、PyTorch和TensorFlow等主流ML的模型转换。对于某些不支持的算子，替换成等效的自定义算子完成模型转换。在本篇文章中，我们以Pytorch 版本的MobilenetV3为案例，介绍如何使用算子替换的方法转换模型。
torch.randperm 不随机
最新发布
01-31
### 解决 `torch.randperm` 不随机的问题 当遇到 `torch.randperm` 的行为不符合预期的情况时，可能的原因之一是伪随机数生成器 (PRNG) 被初始化的方式。为了确保每次调用 `torch.randperm` 都能获得不同的排列结果...
matlab某数组中产生不重复随机数，数组的随机排序，randperm函数numel，X=X(A)函数
mahara的博客
10-17 5005
例如：A=[1,4,5,8,14,15]，从数组A中产生3个不重复的随机数。（当然也可以从A=1:n个数中产生几个不重复随机数）或者（将数组随机排序）。此过程需要利用randperm函数。 1.randperm函数 y = randperm(n)，y是把1到n这些数随机打乱得到的一个数字序列。也可以认为是生成了1到n的不重复的随机排序的n个数。 >> y = randperm(5) ...
Matlab遗传算法的部分映射交叉算子（pmx）源码.7z
07-12
在遗传算法（Genetic Algorithm, GA）中，交叉算子是优化过程中的一种核心操作，它模拟了生物进化中的基因重组过程。部分映射交叉算子（Partially Mapped Crossover, PMX）是一种广泛应用的交叉策略，尤其适用于解决...
randperm
05-25
randperm 函数是 MATLAB 中的一个随机排列函数，用于生成一个指定范围内随机排列的整数序列，例如： ```matlab p = randperm(n) ``` 其中，n 为整数，表示生成的随机排列整数序列的长度，p 为一个 1*n 的行向量，...
ncnn 部分算子不支持的替换操作
程序猿的视界
04-15 2079
Torch Onnx ncnn F.interpolate resize interp Repeat expand 不支持 Mul/add mul/add BinaryOp ...
Pytorch基础——torch.randperm
热门推荐
gy笨瓜的博客
10-12 3万+
torch.randperm(n)：将0~n-1（包括0和n-1）随机打乱后获得的数字序列，函数名是random permutation缩小 【sample】 torch.randperm(10) ===> tensor([2, 3, 6, 7, 8, 9, 1, 5, 0, 4])
pytorch转onnx相关问题
Mr_health的博客
07-23 1837
这些问题是在转谱归一化spectral_norm中遇到的。 首先遇到的就是torch.mv算子和torch.dot算子不支持的问题。 目前pytorch已经官方实现了谱归一化：spectral_norm，其中包含了torch.mv、 torch.dot算子，转onnx会出现错误 解决办法：将torch.mv和torch.dot用torch.matmul代替，不过可能需要自己改变一下tensor的维度。（通过unsqueeze之类的） 我再解决了上述两个算子后，能够跑torch.onnx.export
【onnx】——pytorch转onnx 手动添加不支持的算子
怡宝2号
01-21 2196
onnx，pytorch
当OpenVINO不支持你所部署模型算子时，你该如何做？
英特尔开发人员专区
11-27 6971
在本文中，我们结合一个模型部署案例，向大家讲解了使用OpenVINO时出现算子不匹配等问题时，对于我们使用者来说，如何在官方开发者的帮助下，解决自己所遇到的问题。另外，通过该方式，我们也向OpenVINO反馈了代码中存在的Bug，在解决我们问题的同时，也积极为开源做出了自己的贡献。
模型部署——CenterPoint转ONNX(自定义onnx算子)
xiaohu的博客
10-27 4236
4.torch stack算子 onnx不支持，导出onnx计算图很，将stack和后续PointPillarScatter转换一起定义为ScatterBevPlugin算子，自定义onnx节点和自定义TensorRT算子来实现。2.torch_scatter的scatter_mean和scatter_max onnx不支持，人为自定义onnx节点，后续并自定义tensorRT的scatter_mean和scatter_max算子。
pytorch---onnx---tensorflow踩坑笔记
justsolow的博客
12-31 3203
pytorch—onnx—tensorflow踩坑笔记 首先我转的模型是centertrack这个模型，其实总的来说这个模型还是比较简单的，但是由于其中有一个DCN卷积在onnx和tensorflow中不支持的自定义算子，所以有很多坑都是围绕他进行的。 首先就是对这个DCN卷积部分的处理，我使用了一个插件的形式来方便插入到onnx中，这是由于onnx支持自定义算子的实现。 插件部分：（类似于这样的插件定义的形式） ###############################################
Unsupported ONNX opset version: 11
jacke121的专栏
02-25 1万+
Unsupported ONNX opset version: 11 估计是pytorch不支持，重新安装pytorch2.0可能能解决。
ONNX支持算子列表
tugouxp的专栏
03-26 4815
ONNX算子列表 Abs-13 SequenceEmpty Abs-6 SequenceErase Abs-1 SequenceInsert Acos-7 SequenceLength Acosh-9 Shape Add-14 Shrink Add-13 Sigmoid Add-7 Sign Add-6 Sin Add-1 Sinh And-7 Slice And-1 ...
导出onnx模型，onnx不支持roll操作符
weixin_44791757的博客
04-26 3330
当我torch.onnx.export（）函数导出swin-tranformer模型时出现报错: Exporting the operator roll to ONNX opset version 9 is not supported. Please open a bug to request ONNX export support for the missing operator. 解决方法： 将opset version 9换成12（可以先试试只换opset版本，如果其报错不支持这个版本后
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

曙光_deeplove

你的鼓励是我努力的最大源泉

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值