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实现简单QQ单用户机器人的方法
Jul 03 Python
使用Python压缩和解压缩zip文件的教程
May 06 Python
python实现的守护进程(Daemon)用法实例
Jun 02 Python
python高手之路python处理excel文件(方法汇总)
Jan 07 Python
一条命令解决mac版本python IDLE不能输入中文问题
May 15 Python
Django 浅谈根据配置生成SQL语句的问题
May 29 Python
Python Excel处理库openpyxl使用详解
May 09 Python
python实现批量视频分帧、保存视频帧
May 31 Python
django页面跳转问题及注意事项
Jul 18 Python
Python timeit模块的使用实践
Jan 13 Python
使用 django orm 写 exists 条件过滤实例
May 20 Python
手把手教你实现PyTorch的MNIST数据集
Jun 28 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 xml-rpc远程调用
2008/12/19 PHP
php中的比较运算符详解
2013/10/28 PHP
PHP+mysql实现的三级联动菜单功能示例
2019/02/15 PHP
iis6+javascript Add an Extension File
2007/06/13 Javascript
Knockoutjs的环境搭建教程
2012/11/26 Javascript
javascript实现焦点滚动图效果 具体方法
2013/06/24 Javascript
js禁止页面刷新禁止用F5键刷新禁止右键的示例代码
2013/09/23 Javascript
js为空或不是对象问题的快速解决方法
2013/12/11 Javascript
textarea 控制输入字符字节数(示例代码)
2013/12/27 Javascript
javascript中一些util方法汇总
2015/06/10 Javascript
Google 地图API Map()构造器详解
2016/08/06 Javascript
Bootstrap在线电子商务网站实战项目5
2016/10/14 Javascript
利用JQuery直接调用asp.net后台的简单方法
2016/10/27 Javascript
基于javascript实现按圆形排列DIV元素(一)
2016/12/02 Javascript
vue绑定的点击事件阻止冒泡的实例
2018/02/08 Javascript
一次记住JavaScript的6个正则表达式方法
2018/02/22 Javascript
Angular刷新当前页面的实现方法
2018/11/21 Javascript
vue实现数字动态翻牌的效果(开箱即用)
2019/12/08 Javascript
Vue中import from的来源及省略后缀与加载文件夹问题
2020/02/09 Javascript
Javascript实现秒表计时游戏
2020/05/27 Javascript
利用PHP实现递归删除链表元素的方法示例
2020/10/23 Javascript
使用Python绘制图表大全总结
2017/02/11 Python
Windows下安装python MySQLdb遇到的问题及解决方法
2017/03/16 Python
Python实现的朴素贝叶斯分类器示例
2018/01/06 Python
python实现跨excel的工作表sheet之间的复制方法
2018/05/03 Python
Python3.7在anaconda里面使用IDLE编译器的步骤详解
2020/04/29 Python
如何使用Python处理HDF格式数据及可视化问题
2020/06/24 Python
html特殊符号示例 html特殊字符编码对照表
2014/01/14 HTML / CSS
详解HTML5常用的语义化标签
2019/09/27 HTML / CSS
广告学专业应届生求职信
2013/10/01 职场文书
领导的自我鉴定
2013/12/28 职场文书
行政部工作岗位职责范本
2014/03/05 职场文书
献爱心捐款倡议书
2014/05/14 职场文书
学生检讨书范文
2014/10/30 职场文书
毕业论文致谢格式模板
2015/05/14 职场文书
详解Flutter自定义应用程序内键盘的实现方法
2022/06/14 Java/Android