mindspore中的pic.transpose使用问题

该博客探讨了如何将PyTorch中的数组转换方法应用于Mindspore。torch.from_numpy()在Mindspore中对应为Tensor.from_numpy(),而pic.transpose([3,0,1,2])在Mindspore中可以通过Tensor.transpose()或使用ops.Transpose算子实现。建议查阅Mindspore API文档以获取准确的代码实现。
摘要由CSDN通过智能技术生成

问题描述:

def video_to_tensor(pic):

   return torch.from_numpy(pic.transpose([3,0,1,2]))

看touch代码时遇到这个转化数组的,如果要将其转化为mindspore的形式,torch.from_numpy可以对于Tensor.from_numpy, 但pic.transpore怎么用mindspore表示呢?

解答:

直接使用Tensor的方法:

https://mindspore.cn/docs/api/zh-CN/r1.6/api_python/mindspore/mindspore.Tensor.html?highlight=transpose#mindspore.Tensor.transpose

使用Transpose算子:

https://mindspore.cn/docs/api/zh-CN/r1.6/api_python/ops/mindspore.ops.Transpose.html?highlight=transpose#mindspore.ops.Transpose

建议查询下官网的api文档(注意使用与mindspore版本对应的api文档,界面上可以选择版本号)

为啥torch的脚本可以用np,mindspore的脚本就不能继续用了,需要调整写法。 如果pic是个np,pic.transpose的结果也是np,这里的处理保持原样就可以,把结果封装成mindspore的tensor就好了;

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值