将pytorch转成longtensor的简单方法
更新时间:2020年02月18日 09:00:31 作者:昕晴
今天小编就为大家分享一篇将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的简单方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Python Matplotlib绘制箱型图(箱线图)boxplot的方法详解
箱线图(箱型图)主要作用是发现数据内部整体的分布分散情况,包括上下限、各分位数、异常值等,本文为大家整理了Matplotlib绘制箱型图的所以方法,希望对大家有所帮助2023-05-05Python利用subplots_adjust方法解决图表与画布的间距问题
这篇文章主要介绍了如何在使用python 的 matplotlib库绘图时, 使用subplots_adjust()方法来调整图表与画布之间的间距,以及图表与图表之间的间距,感兴趣的可以了解一下2022-04-04
最新评论