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
0
投稿猜你喜欢
针对之前win10下安装mysql的笔记进行了总结，分享给大家。1. 解压mysql-5.7.11-winx64.zip 到某文件夹， 如C- Pygame是一个多用于游戏开发的模块。本文实例主要是在演示框里实现一个移动的矩形实例代码，完整代码如下：#moving rectangle
- 准备在断网的和联网的机器安装pip，下载地址https://pypi.python.org/pypi/pip在联网的开发机器上安装好需要的包
- 之前写了Python实现登录接口的示例代码，最近需要回顾，就顺便发到随笔上了要求：1.输入用户名和密码2.认证成功，显示欢迎信息3.用户名3
- 在我们人生的路途中，找工作是每个人都会经历的阶段，小编曾经也是苦苦求职大军中的一员。怀着对以后的规划和想象，我们在找工作的时候，会看一些招聘
- 本文实例讲述了Python双精度浮点数运算并分行显示操作。分享给大家供大家参考，具体如下：#coding=utf8def doubleTyp
- 先看一个示例代码如下<template> <div align="center" styl
- 本文实例讲述了Python实现的求解最大公约数算法。分享给大家供大家参考，具体如下：使用Python求解两个数的最大公约数的时候用到了前面介
- pymysql写入中文直接写入中文会出现错误。根据网上的方法，设置my.ini不成功sql = "INSERT INTO comp
- 1 序言1.1 生存压力带来的哲思马尔萨斯最早发现，生物按照几何级数高度增殖的天赋能力，总是大于他们的实际生存能力或现实生存群量，依次推想，
- 本文实例讲述了PHP基于phpqrcode类生成二维码的方法。分享给大家供大家参考，具体如下：使用PHP语言生成二维码，还是挺有难度的，当然
- 一、使用ddt和data装饰器的大致框架如下，每个test_开头的方法，代表一条测试用例from ddt import ddt,dataim
- 什么是类，对象，实例，类## 标题变量（类属性），实例变量（实例属性）面向对象编程（Object-oriented Programming，
- 前言：在自动化测试中，经常需要查找操作文件，比如说查找配置文件（从而读取配置文件的信息），查找测试报告（从而发送测试报告邮件），经常要对大量
- FFmpeg简介FFmpeg是一套可以用来记录、转换数字音频、视频，并能将其转化为流的开源计算机程序。采用LGPL或GPL许可证。它提供了录
- 本文实例为大家分享了JavaScript实现简单计算器的具体代码，供大家参考，具体内容如下代码：<!DOCTYPE html>&
- 温馨提示，只想看图的画直接跳到最后一节拉格朗日方程此前所做的一切三体和太阳系的动画，都是基于牛顿力学的，而且直接对微分进行差分化，从而精度非
- 1005：创建表失败1006：创建数据库失败1007：数据库已存在，创建数据库失败1008：数据库不存在，删除数据库失败1009：不能删除数
- 前一段时间有发过一个简单的JMAIL邮件发邮件的代码，今天就把这个代码做一个具体的注解，并增加了另外两个格式的代码，并举几个简单
- 1 输出大写字母、小写字母、大小写字母、数字、大小写字母和数字1.1输出小写：找到小写a（97）到z（122）的的ASCII码，然后转义为字
