tensor数据类型转换_PyTorch的tensor数据类型及其相关转换

2024-05-13 04:11:30

1 基本类型

PyTorch的数据类型为各式各样的tensor，tensor可以理解为高维矩阵，与Numpy中的array类似。

PyTorch中的tensor包括CPU上的数据类型和GPU上的数据类型。

Torch定义了七种CPU tensor类型和八种GPU tensor类型。

通过使用type函数可以查看变量类型，即当data为tensor数据类型时，使用data.type()可以得到data的数据类型。

更多信息参考

torch.Tensor - PyTorch中文文档pytorch-cn.readthedocs.iotorch.Tensor - PyTorch 1.7.0 documentationpytorch.org

2 类型转换

2.1 CPU tensor和GPU tensor之间的转换

（1）从CPU tensor到GPU tensor，使用data.cuda()。

（2）从GPU tensor到CPU tensor，使用data.cpu()。

2.2 Tensor与Numpy array之间的转换

（1）Tensor到Numpy array可以使用data.numpy()，其中data的类型为torch.Tensor。

（2）Numpy array到Tensor可以使用torch.from_numpy(data)，其中data的类型为numpy.ndarray。

2.3 CPU tensor之间的转换或GPU tensor之间的转换

（1）一般只要在tensor后加long()，int()，double()，float()，byte()等函数就能将tensor进行类型转换。

例如：Torch.LongTensor转换为Torch.FloatTensor，直接使用data.float()即可。

（2）还可以使用type()函数。

当data为tensor数据类型，如果使用data.type(torch.FloatTensor)则强制转换data为torch.FloatTensor类型张量。

（3）当不知道要转换为什么数据类型，但需要求a1，a2两个张量的乘积时，可以使用a1.type_as(a2)将a1转换为a2同类型。

tensor数据类型转换_PyTorch的tensor数据类型及其相关转换相关推荐

判定两个tensor维度相同_Pytorch之Tensor大详解
导语深度学习框架Pytorch发展势头惊人,这点小编也深有体会,翻翻Github上深度学习的开源代码,发现用Pytorch真的多了不少,所以小编最近也正在入坑Pytorch,顺便写写文章做些总结.认 ...
前端：JS/17/前篇总结（JS程序的基本语法，变量），数据类型-变量的类型（数值型，字符型，布尔型，未定义型，空型），数据类型转换，typeof()判断数据类型，从字符串提取整数或浮点数的函数
JS程序的基本语法 1,JS是区分大小写的,如:Name和name是不同的: 2,JS中每一条语句,一般以英文的分号(;)结束,在Java中是严格地一定要加上分号,不然会报错:但在JavaScript ...
python选取tensor某一维_Pytorch的Tensor操作(1)
类型推断 torch.randn():随机初始化 a.type():返回类型 type():返回基本类型 isinstance() :检查类型 cuda会影响数据类型标量维度(dimention) ...
pyspark 数据类型转换_PySpark之DataFrame的创建与转换
简介 DataFrame结构代表的是数据的一个不可变分布式集合,其数据都被组织到有名字的列中,就像关系型数据库中的表一样.DataFrame 的目的就是要让对大型数据集的处理变得更简单,它让开发者可以 ...
06_2_Pytorch的基础数据类型、CPU tensor类型和GPU tensor类型、判断数据类型、CPU或GPU张量之间的转换、数据类型转换、1-4维向量、Tensor张量、Variable等
1.6.2.Pytorch的基础数据类型 1.6.2.1.Torch定义了的七种CPU tensor类型和八种GPU tensor类型 1.6.2.2.基础数据类型 1.6.2.3.Pytorch数据 ...
python基础篇1(print函数，转义字符，保留字和标识符，变量，数据类型和数据类型转换)
文章目录 print()函数转义字符 python中的标识符和保留字变量的定义和使用什么是变量变量的赋值数据类型整数类型浮点数布尔类型字符串类型数据类型转换 print()函数描 ...
数据类型转换pytorch
du = torch.ones([2,2]) a = np.array([[1,2],[3,4]],dtype=np.float32) b = torch.from_numpy(a)#数据类型是不变的 ...
Pytorch数据类型转换
转自:https://blog.csdn.net/weixin_40446557/article/details/88221851 1.Pytorch上的数据类型 Pytorch的类型可以分为CPU和 ...
MYSQL 与 Oracle 之间的数据类型转换
2009-07-21 MYSQL 与 Oracle 之间的数据类型转换关键字: mysql, oracle, 数据类型, 转换 Table 2-4 Default Data Type Mapping ...

最新文章

热门文章