如何将PyTorch中的数据转换为LongTensor的简便步骤？

本文共计362个文字，预计阅读时间需要2分钟。

pythonimport torch

创建一个随机长为3x3的张量b=torch.rand(3, 3)

我就废话不多说了，直接上代码了。非常简单哦！

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的简单方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持易盾网络。