torch转onnx时,遇到 linspace操作不支持的问题

Exporting the operator linspace to ONNX opset version 11 is not supported.

torch.linspace貌似不支持转onnx

查阅资料后,用
torch.arange 来代替了,
但需要注意

torch.arange(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

arrange返回的是 [start, end)之间步长为

在这里插入图片描述
的一维向量。且需要注意,返回的数组是大于等于start,小于end

o u t i + 1 = o u t i + s t e p out_{i+1}=out_i+step outi+1=outi+step

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值