pytorch实现Tensor变量之间的转换
作者:zhe_csdn 发布时间:2023-06-28 15:41:31
系统默认是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)
降维也是做同样的操作
来源:https://blog.csdn.net/zhe_csdn/article/details/93655053
猜你喜欢
- Usage example (libtiff wrapper)from libtiff import TIFF# to open a tif
- PHP保存session默认的是采用的文件的方式来保存的,这仅仅在文件的空间开销很小的windows上是可以采用的,但是如果我们采用uinx
- 本文介绍了asp中 adpbe.stream 的语法,各种参数使用说明,方便大家查阅。更多请看:VBScript 速查手册(语言参考) ch
- 如下所示:def trans_data_to_pair(self,data,index): contents=[
- 可以使用 Application 对象在给定的应用程序的所有用户之间共享信息。基于 ASP 的应用程序同所有的 .asp 文件一样在一个虚拟
- Python版本 实现了比之前的xxftp更多更完善的功能1、继续支持多用户2、继续支持虚拟目录3、增加支持用户根目录以及映射虚拟目录的权限
- gRPC 是可以在任何环境中运行的现代开源高性能 RPC 框架。它可以通过可插拔的支持来有效地连接数据中心内和跨数据中心的服务,以实现负载平
- 为了应用方便,您可能需要给数据库的每条记录都添加日期/时间戳,以便确定各个记录添加到数据库的时间。在Access数据库应用中,使用Now()
- 本文实例讲述了python实现生成Word、docx文件的方法。分享给大家供大家参考,具体如下:http://python-docx.rea
- 这篇文章主要介绍了python可视化text()函数使用详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需
- Pytorch 使用GPU训练使用 GPU 训练只需要在原来的代码中修改几处就可以了。我们有两种方式实现代码在 GPU 上进行训练方法一 .
- 概率论啊概率论,差不多忘完了。基于概率论的分类方法:朴素贝叶斯1. 概述贝叶斯分类是一类分类算法的总称,这类算法均以贝叶斯定理为基础,故统称
- SQL Server 2008的独到之处:安装SQL Server 2008的设置和安装也有所改进。配置数据和引擎位已经分开了,所以它使创建
- 废话不多说,看代码吧!'''待完善。此代码实现了,根据标注文本的属性,数值,位置,及 容差,去判断 设计 和 实测两
- 1. Python字典的clear()方法(删除字典内所有元素)#!/usr/bin/python# -*- coding: UTF-8 -
- 用HZHOST实用工具集的服务器安全设置里安装了MSSQL安全配置,现在SQL2000还原不了数据库了,从还原选定设备浏览文件夹时出现&qu
- 前面已经介绍了关于Dreamweaver MX 2004的基本操作
- 对于变量的访问和设置,我们可以使用get、set方法,如下:class student: def __init__(self,n
- 常用的python第三方库安装工具大概有三种:1、pip (分为cmd和conda)(推荐)2、easy_install &n
- 函数:string.join()Python中有join()和os.path.join()两个函数,具体作用如下:join(): 连接字符串