我就废话不多说了，直接上代码了。非常简单哦！
pytorch转成longtensor
b = torch.rand(3,3)#得到的是floattensor值，b = b.long()#得到的是longtensor值pytorch tensor转换实例
import torchimport numpy as np #(1,3,16,2,2)# 1个视频，16个关键帧，3通道，长为2宽为2# 即一个视频，有16张 3通道，长为2宽为2 的图像data = np.array([[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3] ,[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3] ,[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3] ,[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3]]) print data torch_data = torch.FloatTensor(data) # 转tensorprint torch_data b = torch_data.view(1,3,16,2,2)print b以上这篇将pytorch转成longtensor的简单方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。
最新文章
热门文章
猜你喜欢
