将pytorch转成longtensor的简单方法
6 浏览量
2020-09-17
21:54:06
上传
评论
收藏 29KB PDF 举报
将将pytorch转成转成longtensor的简单方法的简单方法
今天小编就为大家分享一篇将pytorch转成longtensor的简单方法，具有很好的参考价值，希望对大家有所帮
助。一起跟随小编过来看看吧
我就废话不多说了，直接上代码了。非常简单哦！
pytorch转成转成longtensor
b = torch.rand(3,3)#得到的是floattensor值，
b = b.long()#得到的是longtensor值
pytorch tensor转换实例转换实例
import torch
import 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) # 转tensor
print torch_data
b = torch_data.view(1,3,16,2,2)
print b
以上这篇将pytorch转成longtensor的简单方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支
持我们。
评论0
最新资源