pytorch常用数据类型所占字节数对照表一览


Posted in Python onMay 17, 2021

PyTorch上的常用数据类型如下

Data type dtype CPU tensor GPU tensor Size/bytes
32-bit floating torch.float32 or torch.float torch.FloatTensor torch.cuda.FloatTensor 4
64-bit floating torch.float64 or torch.double torch.DoubleTensor torch.cuda.DoubleTensor 8
16-bit floating torch.float16or torch.half torch.HalfTensor torch.cuda.HalfTensor -
8-bit integer (unsigned) torch.uint8 torch.ByteTensor torch.cuda.ByteTensor 1
8-bit integer (signed) torch.int8 torch.CharTensor torch.cuda.CharTensor -
16-bit integer (signed) torch.int16or torch.short torch.ShortTensor torch.cuda.ShortTensor 2
32-bit integer (signed) torch.int32 or torch.int torch.IntTensor torch.cuda.IntTensor 4
64-bit integer (signed) torch.int64 or torch.long torch.LongTensor torch.cuda.LongTensor 8

以上PyTorch中的数据类型和numpy中的相对应,占用字节大小也是一样的

补充:pytorch tensor比较大小 数据类型要注意

如下

a = torch.tensor([[0, 0], [0, 0]])
print(a>=0.5)

输出

tensor([[1, 1],

[1, 1]], dtype=torch.uint8)

结果明显不对, 分析原因是因为, a是long类型, 而0.5是float. 0.5会被转化为 long, 变为0. 因此结果会出错, 做出如下修改就可以得到正确答案

正确用法:

a = torch.tensor([[0, 0], [0, 0]]).float()
print(a>=0.5)

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中logging模块的用法实例
Sep 29 Python
python爬虫之BeautifulSoup 使用select方法详解
Oct 23 Python
python自动12306抢票软件实现代码
Feb 24 Python
python使用matplotlib库生成随机漫步图
Aug 27 Python
Pycharm无法使用已经安装Selenium的解决方法
Oct 13 Python
python 一个figure上显示多个图像的实例
Jul 08 Python
Python中的With语句的使用及原理
Jul 29 Python
搭建pypi私有仓库实现过程详解
Nov 25 Python
python3中celery异步框架简单使用+守护进程方式启动
Jan 20 Python
Python机器学习工具scikit-learn的使用笔记
Jan 28 Python
python 通过使用Yolact训练数据集
Apr 06 Python
python manim实现排序算法动画示例
Aug 14 Python
python使用tkinter实现透明窗体上绘制随机出现的小球(实例代码)
Python编写可视化界面的全过程(Python+PyCharm+PyQt)
Pytorch 实现变量类型转换
Python进度条的使用
May 17 #Python
Python包管理工具pip的15 个使用小技巧
Python中json.dumps()函数的使用解析
May 17 #Python
Python中threading库实现线程锁与释放锁
You might like
php在字符串中查找另一个字符串
2008/11/19 PHP
为PHP初学者的8点有效建议
2010/11/20 PHP
php实现邮件发送并带有附件
2014/01/24 PHP
WebQQ最新登陆协议的用法
2014/12/22 PHP
php封装好的人民币数值转中文大写类
2015/12/20 PHP
Laravel 5.4向IoC容器中添加自定义类的方法示例
2017/08/15 PHP
Mootools 1.2教程 同时进行多个形变动画
2009/09/15 Javascript
jquery点击页面任何区域实现鼠标焦点十字效果
2013/06/21 Javascript
event对象获取方法总结在google浏览器下测试
2013/11/03 Javascript
JavaScript中对象属性的添加和删除示例
2014/05/12 Javascript
jquery通过load获取文件的内容并跳到锚点的方法
2015/01/29 Javascript
js淡入淡出的图片轮播效果代码分享
2015/08/24 Javascript
轻松掌握jQuery中wrap()与unwrap()函数的用法
2016/05/24 Javascript
javascript 数组的正态分布排序的问题
2016/07/31 Javascript
你真的了解BOM中的history对象吗
2017/02/13 Javascript
vue-cli的eslint相关用法
2017/09/29 Javascript
JS同步、异步、延迟加载的方法
2018/05/05 Javascript
Vue + better-scroll 实现移动端字母索引导航功能
2018/05/07 Javascript
Element-ui tree组件自定义节点使用方法代码详解
2018/09/17 Javascript
vue实现页面内容禁止选中功能,仅输入框和文本域可选
2019/11/09 Javascript
d3.js 地铁轨道交通项目实战
2019/11/27 Javascript
详解js创建对象的几种方式和对象方法
2021/03/01 Javascript
[00:52]黑暗之门更新 新英雄孽主驾临DOTA2
2016/08/24 DOTA
Python库urllib与urllib2主要区别分析
2014/07/13 Python
详解Python Socket网络编程
2016/01/05 Python
python变量不能以数字打头详解
2016/07/06 Python
python对象与json相互转换的方法
2019/05/07 Python
解决yum对python依赖版本问题
2019/07/05 Python
opencv+pyQt5实现图片阈值编辑器/寻色块阈值利器
2020/11/13 Python
shell程序如何生命变量?shell变量是弱变量吗?
2014/11/10 面试题
运动会100米解说词
2014/01/23 职场文书
企业家王石演讲稿:坚持与放下
2014/04/27 职场文书
建党伟业观后感
2015/06/01 职场文书
生产实习心得体会范文
2016/01/22 职场文书
golang 实现时间戳和时间的转化
2021/05/07 Golang
python读取mnist数据集方法案例详解
2021/09/04 Python