将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的简单方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
PIL是一个用于图像处理的Python库,它提供了各种功能,包括加载、保存、编辑和处理图像,你可以使用PIL库进行图像缩放、裁剪、旋转、滤镜应用等操作,本文将介绍如何使用Python的PIL库给图像进行过滤,需要的朋友可以参考下2023-08-08
Jupyter Notebook内使用argparse报错的解决方案
这篇文章主要介绍了在Jupyter Notebook内使用argparse报错的解决方案，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-06-06
单例模式是一种常用的软件设计模式，该模式的主要目的是确保某一个类只有一个实例存在，本文给大家分享Python 实现单例模式的五种写法，感兴趣的朋友一起看看吧2022-11-11
这篇文章主要为大家详细介绍了Python+OpenCV采集本地摄像头的视频，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-04-04
这篇文章主要为大家详细介绍了Linux终端上python实现简易聊天室功能，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-07-07
这篇文章主要介绍了Keras 使用 Lambda层详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
不同对象比较是按照 type names 比较，当相同类型对象不支持适当比较的时候，采用 address 比较,需要的朋友可以参考下2014-08-08
在现代软件开发中,任务管理是一个至关重要的功能,这篇文章主要为大家介绍了如何使用Python构建一个简单的任务管理器,有需要的可以参考下2024-11-11
这篇文章主要介绍了Pytest之测试命名规则的使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-04-04
今天小编就为大家分享一篇对Python中9种生成新对象的方法总结，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
最新评论