[Pytorch]Pytorch的tensor变量类型转换

最新推荐文章于 2025-06-18 16:58:08 发布
最新推荐文章于 2025-06-18 16:58:08 发布
阅读量1w 收藏 8
点赞数 2
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/weixin_44538273/article/details/88408971
本文介绍了Pytorch中Tensor的类型转换，包括CPU与GPU间的转换，不同数据类型的转换，以及Variable到Tensor和Tensor到Numpy Array的转换。转换方法包括使用函数如float(), long(), cuda(), cpu(), type_as()等。" 81036417,7424330,长亭科技发布新品洞鉴和牧云，牧云即将开源,"['网络安全', '云安全', '扫描器', '开源软件', '融资']

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

[Pytorch]Pytorch的tensor变量类型转换

原文：https://blog.youkuaiyun.com/hustchenze/article/details/79154139
Pytorch的数据类型为各式各样的Tensor,Tensor可以理解为高维矩阵。与Numpy中的Array类似。Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型，一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。通过使用Type函数可以查看变量类型。一般系统默认是torch.FloatTensor类型。例如data = torch.Tensor(2,3)是一个2*3的张量，类型为FloatTensor; data.cuda()就转换为GPU的张量类型，torch.cuda.FloatTensor类型。

下面简单介绍一下Pytorch中变量之间的相互转换。

（1）CPU或GPU张量之间的转换

一般只要在Tensor后加long(), int(), double(),float(),byte()等函数就能将Tensor进行类型转换；

例如：Torch.LongTensor--->Torch.FloatTensor, 直接使用data.float()即可

还可以使用type

最低0.47元/天 解锁文章

200万优质内容无限畅学
急流勇进
关注
pytorch踩坑记
m0_46545833的博客
03-21 1907
pytorch神经网络
深度学习（二）pytorch入门——tensor
2301_79712009的博客
11-05 1340
在实际的应用中，经常需要增加或减少Tensor的维度，尤其是维度为1的情况，这时候可以使用squeeze()与unsqueeze()函数，前者用于去除size为1的维度，而后者则是将指定的维度的size变为1。()函数可以在不改变Tensor数据的前提下任意改变Tensor的形状，必须保证调整前后的元素总数相同，并且调整前后内存共享，三者的作用基本相同。ranspose()函数可以将指定的两个维度的元素进行转置，而permute()函数可以按照给定的维度进行维度变换。张量的形状表示张量在每个维度上的大小。
Pytorch变量类型转换
热门推荐
zchenack个人专栏
01-24 9万+
Pytorch的数据类型为各式各样的Tensor,Tensor可以理解为高维矩阵。与Numpy中的Array类似。Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型，一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。通过使用Type函数可以查看变量类型。一般系统默认是torch.FloatTensor类型。例如data = torch.Tensor(
PyTorch 中Tensor常用数据结构（int, list, numpy array等）互相转换和实战示例
最新发布
qq_36812406的博客
06-18 729
在 PyTorch 中，tensor是一种强大且灵活的数据结构，可以与多种 Python 常用数据结构（如intlist等）互相转换。
Torch类型转换type、type_as
taoqick的专栏
09-02 1320
【代码】Torch类型转换type、type_as。
torch中的数据类型和相互转换
程西西
12-12 2万+
文章目录1 torch.Tensor2 Data types3 Initializing and basic operations1）使用torch.tensor() 创建2）使用python list创建3）使用zeros ones函数创建4）通过torch.dtype指定类型， torch.device指定构造器5）零维张量6）torch.Tensor.item()从包含单个值的张量中获取Python数字4 type changes1）tensor间类型转换2）数据存储位置转换3）与Python数据类型
【Torch tensor】torch中的数据类型和相互转换
devil_son1234的博客
02-15 4210
torch中的数据类型和相互转换
pytorch实现Tensor变量之间的转换
09-17
对于Tensor的类型转换，可以通过在Tensor后面添加相应的类型转换函数来完成： - `.long()`：将Tensor转换为`LongTensor`（整型）。 - `.int()`：将Tensor转换为`IntTensor`（整型）。 - `.double()`：将Tensor转换为...
Pytorch 之修改Tensor部分值方式
09-18
首先，我们需要了解Pytorch的Tensor数据类型。Tensor类似于多维数组，它可以存储多维数据，通常用于存储输入数据、模型参数、中间变量以及输出数据等。与Python的列表相比，Tensor可以利用GPU进行高效计算，这对于...
Pytorch数据类型转换（torch.tensor，torch.FloatTensor）
ggbooo的博客
02-15 2万+
Pytorch数据类型转换（torch.tensor，torch.FloatTensor）
pytorch | tensor转换为float
Mopes__的博客
03-26 1869
pytorch | tensor转换为float
Pytorch数据类型转换
Blankit1的博客
08-20 7460
Pytorch-数据类型转换 ...
Pytorch学习笔记——数据类型转换总结
m0_45388819的博客
11-01 623
1.Tensor与Numpy之间的相互转化 tensor转numpy a = torch.ones([2,5]) b = a.numpy() numpy转tensor a = np.ones([2,5]) b = torch.from_numpy(a) 2.Tensor与list之间的相互转化 tensor转list a = torch.ones([1,5]) b = a.tolist() list转tensor a = list(range(1,6)) b = torch.tensor(a) 3.
pytorch学习笔记——数据类型转化
phily123的博客
11-28 1049
torchvision.transforms.ToTensor(ndarry/PIL) 注意： 这种转化方式会将数据归一化，不过cv2.imshow()显示图片时会自动将归一化的像素值乘以255，自己之前将经过ToTensor归一化的像素值直接乘以255再输入给imshow函数结果都是白色，原来imshow函数内部判断数据类型是float32会自己处理。 对于imshow函数，opencv的官方注释指出：根据图像的深度，imshow函数会自动对其显示灰度值进行缩放，规则如下： 1.如果图像数据类型是8U（.
PyTorch常用函数(5)
weixin_43229348的博客
09-15 1439
1. torch.Storage 一个 torch.Storage 是一个单一数据类型的连续一维数组。每个 torch.Tensor 都有一个对应的、相同数据类型的存储。 byte()：将此存储转为byte类型 char()：将此存储转为char类型 clone()：返回此存储的一个副本 cpu()：如果当前此存储不在 CPU 上，则返回一个它的 CPU 副本 cuda(device=None, async=False)：返回此对象在 CUDA 内存中的一个副本。如果此对象已在 CUDA 内存中且在正确的
pytoch 使用float64训练
gltangwq的博客
06-03 7396
pytorch默认使用单精度float32训练模型，原因在于：使用float16训练模型，模型效果会有损失，而使用double(float64)会有2倍的内存压力，且不会带来太多的精度提升。 本人，最近遇到需要使用double数据类型训练模型的情况，具体实现需要把模型的权重参数数据类型和输入数据类型全部设置为torch.float64即可。 可使用torch的一个函数，轻松地把模型参数转化为float64 torch.set_default_dtype(torch.float64) 输入类型可使用 ten
pytorch 基础
xy52240915的博客
12-13 485
这里写自定义目录标题默认 Variable 是不需要求梯度的，所以我们用这个方式申明需要对其进行求梯度欢迎使用Markdown编辑器新的改变功能快捷键合理的创建标题，有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能...
如何将数据存储为pytorch的tensor变量
05-23
将数据存储为 PyTorch 的 tensor 变量可以使用 `torch.tensor()` 函数。这个函数可以接受多种 Python 的数据类型，如 list、tuple、numpy array 等等。 以下是一个将 numpy array 存储为 PyTorch tensor 的示例代码： ```python import numpy as np import torch # 创建一个 numpy array data = np.array([[1, 2], [3, 4]]) # 将 numpy array 转换为 PyTorch tensor tensor_data = torch.tensor(data) print(tensor_data) ``` 输出： ``` tensor([[1, 2], [3, 4]]) ``` 你也可以在创建 tensor 时指定数据类型（dtype）和设备（device）： ```python # 创建一个 float 类型的 tensor，在 GPU 上存储 tensor_data = torch.tensor(data, dtype=torch.float32, device='cuda') ``` 需要注意的是，将数据存储为 tensor 变量后，该变量可以直接传入到 PyTorch 模型中进行计算。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值