pytorch实现Tensor变量之间的转换
系统默认是torch.FloatTensor类型
data = torch.Tensor(2,3)是一个2*3的张量，类型为FloatTensor
data.cuda()就转换为GPU的张量类型，torch.cuda.FloatTensor类型
(1) CPU或GPU之间的张量转换
在Tensor后加long(), int(), double(),float(),byte()等函数就能将Tensor进行类型转换type()函数，
data为Tensor数据类型，data.type()为给出data的类型，
如果使用data.type(torch.FloatTensor)则强制转换为torch.FloatTensor类型张量
(2) CPU张量转化成GPU张量
data.cuda()
(3) GPU张量转化成CPU张量
data.cpu()
(4) Variable变量转换成普通的Tensor
Variable是一个Wrapper,装在里面的data是tensor，如果Var是Variable变量，使用Var.data获得Tensor变量
(5) Tensor与numpy array之间的转换
Tensor->numpy 使用data.numpy(),data为Tensor变量
Numpy->Tensor 使用torch.from_numpy(data)，data为numpy变量
(6) 分别获取张量和数组的尺寸,注意size的使用
torch 张量
获取张量的尺寸 a.size()
numpy 数组
获取数组的尺寸 b.shape
获取数组中元素的个数: b.size (这里和张量中的属性的size的含义不同)
(7) 升维和降维的问题
unsqueeze(N)升维到第N维
squeeze(N)降维第N维
需要做如下操作:
x = x.unsqueeze(0) 假如x=(3,1080,1920) 操作后 x = (1,3,1080,1920)
降维也是做同样的操作
以上这篇pytorch实现Tensor变量之间的转换就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python安装sklearn模块的方法详解,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-11-11
Python中使用logging模块代替print（logging简明指南）
这篇文章主要介绍了Python中使用logging模块代替print的好处说明,主旨是logging模块简明指南,logging模块的使用方法介绍,需要的朋友可以参考下2014-07-07
前面九篇文章从基础到编写都做了详细的介绍了，第十篇么讲究个十全十美，那么我们就来详细记录一下一个爬虫程序如何一步步编写出来的，各位看官可要看仔细了2014-11-11
Python的 eval() 允许从基于字符串或基于编译代码的输入中计算任意Python表达式。当从字符串或编译后的代码对象的任何输入中动态计算Python表达式时，此函数非常方便。本文将利用eval实现动态地计算数学表达式，需要的可以参考一下2022-09-09
这篇文章主要介绍了如何利用Python+OpenCV实现人打哈欠检测，文中的示例代码讲解详细，对我们学习Python有一定帮助，感兴趣的可以了解一下2022-04-04
这篇文章主要介绍了Python自动扫雷实现方法,提供了Python实现自动扫雷的代码思路并备有注释说明,具有一定参考借鉴价值,需要的朋友可以参考下2015-07-07
首先，在认识metaclass之前，你需要认识下python中的class。python中class的奇怪特性借鉴了smalltalk语言。大多数语言中，classes仅仅是用于描述怎样创建一个对象的代码端。在某种程度上说，python中的class也是这样的。2016-02-02
python 如何用 Hypothesis 来自动化单元测试
这篇文章主要介绍了python 如何用 Hypothesis 来自动化单元测试，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下2021-03-03
这篇文章主要介绍了Python使用numpy模块创建数组操作,结合实例形式分析了Python使用numpy模块实现数组的创建、赋值、修改、打印等相关操作技巧与注意事项,需要的朋友可以参考下2018-06-06
这篇文章主要介绍了python绘图模块matplotlib示例详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
最新评论