网络编程
位置：首页>> 网络编程>> Python编程>> 将pytorch转成longtensor的简单方法

将pytorch转成longtensor的简单方法

作者：昕晴　　发布时间：2021-10-25 02:42:56　

标签：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

来源：https://blog.csdn.net/qq_40210472/article/details/87895327

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com