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 相关文章推荐
python删除文件示例分享
Jan 28 Python
跟老齐学Python之关于循环的小伎俩
Oct 02 Python
python将图片文件转换成base64编码的方法
Mar 14 Python
利用python获取某年中每个月的第一天和最后一天
Dec 15 Python
Python编程实现微信企业号文本消息推送功能示例
Aug 21 Python
Python 逐行分割大txt文件的方法
Oct 10 Python
Python模块搜索路径代码详解
Jan 29 Python
解决python3 urllib 链接中有中文的问题
Jul 16 Python
pytorch permute维度转换方法
Dec 14 Python
如何利用Python分析出微信朋友男女统计图
Jan 25 Python
TensorFlow索引与切片的实现方法
Nov 20 Python
python GUI库图形界面开发之PyQt5滚动条控件QScrollBar详细使用方法与实例
Mar 06 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 addslashes 利用递归实现使用反斜线引用字符串
2013/08/05 PHP
PHP自带函数给数字或字符串自动补齐位数
2014/07/29 PHP
Yii2框架实现利用mpdf创建pdf文件功能示例
2019/02/08 PHP
PHP中抽象类,接口功能、定义方法示例
2019/02/26 PHP
PHP 对象接口简单实现方法示例
2020/04/13 PHP
JavaScript window.setTimeout() 的详细用法
2009/11/04 Javascript
jquery checkbox,radio是否选中的判断代码
2010/03/20 Javascript
js chrome浏览器判断代码
2010/03/28 Javascript
jquery 图片 上一张 下一张 链接效果(续篇)
2010/04/20 Javascript
js数组依据下标删除元素
2015/04/14 Javascript
js实现TAB切换对应不同颜色的代码
2015/08/31 Javascript
jquery+正则实现统一的表单验证
2015/09/20 Javascript
13个PHP函数超实用
2015/10/21 Javascript
JavaScript的ExtJS框架中表格的编写教程
2016/05/21 Javascript
javascript弹出带文字信息的提示框效果
2016/07/19 Javascript
Jquery EasyUI $.Parser
2017/06/02 jQuery
axios发送post请求,提交图片类型表单数据方法
2018/03/16 Javascript
微信小程序云开发实现云数据库读写权限
2019/05/17 Javascript
跟老齐学Python之数据类型总结
2014/09/24 Python
Python实现网站文件的全备份和差异备份
2014/11/30 Python
Python下的twisted框架入门指引
2015/04/15 Python
pandas DataFrame数据转为list的方法
2018/04/11 Python
利用python脚本如何简化jar操作命令
2019/02/24 Python
django 微信网页授权登陆的实现
2019/07/30 Python
解决Keras中CNN输入维度报错问题
2020/06/29 Python
使用CSS3实现多列布局与多背景的技巧
2016/02/29 HTML / CSS
详解canvas绘图时遇到的跨域问题
2018/03/22 HTML / CSS
2014新课程改革心得体会
2014/03/10 职场文书
经典英文广告词
2014/03/18 职场文书
财务内勤岗位职责
2014/04/17 职场文书
我爱我班主题班会
2015/08/13 职场文书
2016婚礼主持词开场白
2015/11/24 职场文书
高一作文之乐趣
2019/11/21 职场文书
python opencv常用图形绘制方法(线段、矩形、圆形、椭圆、文本)
2021/04/12 Python
Python 制作自动化翻译工具
2021/04/25 Python
Matlab如何实现矩阵复制扩充
2021/06/02 Python