问题描述:
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就好了;