pytorch中tensor张量数据类型的转化方式


Posted in Python onDecember 31, 2019

1.tensor张量与numpy相互转换

tensor ----->numpy

import torch
a=torch.ones([2,5])

tensor([[1., 1., 1., 1., 1.],
    [1., 1., 1., 1., 1.]])
# **********************************    
b=a.numpy()

array([[1., 1., 1., 1., 1.],
    [1., 1., 1., 1., 1.]], dtype=float32)
numpy ----->tensor

import numpy as np
a=np.ones([2,5])

array([[1., 1., 1., 1., 1.],
    [1., 1., 1., 1., 1.]])
# **********************************    
b=torch.from_numpy(a)

tensor([[1., 1., 1., 1., 1.],
    [1., 1., 1., 1., 1.]], dtype=torch.float64)

2.tensor张量与list相互转换

tensor—>list

a=torch.ones([1,5])

tensor([[1., 1., 1., 1., 1.]])
# ***********************************
b=a.tolist()

[[1.0, 1.0, 1.0, 1.0, 1.0]]



list—>tensor

a=list(range(1,6))

[1, 2, 3, 4, 5]
# **********************************
b=torch.tensor(a)

tensor([1, 2, 3, 4, 5])

3.tensor张量见类型转换

构建一个新的张量,你要转变成不同的类型只需要根据自己的需求选择即可

tensor = torch.Tensor(3, 5)

# torch.long() 将tensor投射为long类型
newtensor = tensor.long()

# torch.half()将tensor投射为半精度浮点类型
newtensor = tensor.half()

# torch.int()将该tensor投射为int类型
newtensor = tensor.int()

# torch.double()将该tensor投射为double类型
newtensor = tensor.double()

# torch.float()将该tensor投射为float类型
newtensor = tensor.float()

# torch.char()将该tensor投射为char类型
newtensor = tensor.char()

# torch.byte()将该tensor投射为byte类型
newtensor = tensor.byte()

# torch.short()将该tensor投射为short类型
newtensor = tensor.short()

4.type_as() 将张量转换成指定类型张量

>>> a=torch.Tensor(2,5)
>>> a
tensor([[1.9431e-19, 4.8613e+30, 1.4603e-19, 2.0704e-19, 4.7429e+30],
    [1.6530e+19, 1.8254e+31, 1.4607e-19, 6.8801e+16, 1.8370e+25]])
>>> b=torch.IntTensor(1,2)
>>> b
tensor([[16843009,    1]], dtype=torch.int32)
>>> a.type_as(b)
tensor([[     0, -2147483648,      0,      0, -2147483648],
    [-2147483648, -2147483648,      0, -2147483648, -2147483648]],
    dtype=torch.int32)
>>> a
tensor([[1.9431e-19, 4.8613e+30, 1.4603e-19, 2.0704e-19, 4.7429e+30],
    [1.6530e+19, 1.8254e+31, 1.4607e-19, 6.8801e+16, 1.8370e+25]])

以上这篇pytorch中tensor张量数据类型的转化方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
centos系统升级python 2.7.3
Jul 03 Python
基于python编写的微博应用
Oct 17 Python
Python合并字符串的3种方法
May 21 Python
Python中遍历字典过程中更改元素导致异常的解决方法
May 12 Python
Python及PyCharm下载与安装教程
Nov 18 Python
python中使用ctypes调用so传参设置遇到的问题及解决方法
Jun 19 Python
python 猴子补丁(monkey patch)
Jun 26 Python
tensorflow 实现自定义layer并添加到计算图中
Feb 04 Python
TensorFlow2.X结合OpenCV 实现手势识别功能
Apr 08 Python
Django微信小程序后台开发教程的实现
Jun 03 Python
Python爬虫自动化爬取b站实时弹幕实例方法
Jan 26 Python
python爬取豆瓣电影排行榜(requests)的示例代码
Feb 18 Python
Pytorch之parameters的使用
Dec 31 #Python
使用TensorFlow-Slim进行图像分类的实现
Dec 31 #Python
Pytorch之view及view_as使用详解
Dec 31 #Python
window环境pip切换国内源(pip安装异常缓慢的问题)
Dec 31 #Python
如何基于Python创建目录文件夹
Dec 31 #Python
Pytorch之contiguous的用法
Dec 31 #Python
python实现将json多行数据传入到mysql中使用
Dec 31 #Python
You might like
优化PHP代码的53条建议
2008/03/27 PHP
PHP实现执行外部程序的方法详解
2017/08/17 PHP
纯CSS3实现质感细腻丝滑按钮
2021/03/09 HTML / CSS
JS 继承实例分析
2008/11/04 Javascript
jQuery ajax cache缓存问题
2010/07/01 Javascript
js 时间函数应用加、减、比较、格式转换的示例代码
2013/08/23 Javascript
使用jQuery插件创建常规模态窗口登陆效果
2013/08/23 Javascript
js图片预加载示例
2014/04/30 Javascript
HTML页面登录时的JS验证方法
2014/05/28 Javascript
理解javascript对象继承
2016/04/17 Javascript
js实现可键盘控制的简单抽奖程序
2016/07/13 Javascript
nodejs密码加密中生成随机数的实例代码
2017/07/17 NodeJs
详解Webpack+Babel+React开发环境的搭建的方法步骤
2018/01/09 Javascript
layui点击按钮添加可编辑的一行方法
2018/08/15 Javascript
Three.JS实现三维场景
2018/12/30 Javascript
JQuery判断radio单选框是否选中并获取值的方法
2019/01/17 jQuery
基于Node.js搭建hexo博客过程详解
2019/06/25 Javascript
Vue组件模板的几种书写形式(3种)
2020/02/19 Javascript
jQuery开发仿QQ版音乐播放器
2020/07/10 jQuery
讲解Python中if语句的嵌套用法
2015/05/14 Python
对Python实现简单的API接口实例讲解
2018/12/10 Python
对python:threading.Thread类的使用方法详解
2019/01/31 Python
Flask教程之重定向与错误处理实例分析
2019/08/01 Python
简单易懂Pytorch实战实例VGG深度网络
2019/08/27 Python
基于python实现文件加密功能
2020/01/06 Python
Python实现Wordcloud生成词云图的示例
2020/03/30 Python
vscode写python时的代码错误提醒和自动格式化的方法
2020/05/07 Python
python实现图片转字符画
2021/02/19 Python
CSS3+DIV实现漂亮的动画彩色标签
2016/06/16 HTML / CSS
运动会演讲稿100字
2014/08/25 职场文书
广告设计专业毕业生自我鉴定
2014/09/27 职场文书
四川省传达学习贯彻党的群众路线教育实践活动总结大会精神新闻稿
2014/10/26 职场文书
汽车销售助理岗位职责
2015/04/14 职场文书
七年级作文之关于奶奶
2019/10/29 职场文书
8g内存用python读取10文件_面试题-python 如何读取一个大于 10G 的txt文件?
2021/05/28 Python
解决vue中provide inject的响应式监听
2022/04/19 Vue.js