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之折腾一下目录
Oct 24 Python
基于python脚本实现软件的注册功能(机器码+注册码机制)
Oct 09 Python
Python求一批字符串的最长公共前缀算法示例
Mar 02 Python
用pyqt5 给按钮设置图标和css样式的方法
Jun 24 Python
python使用原始套接字发送二层包(链路层帧)的方法
Jul 22 Python
numpy.transpose()实现数组的转置例子
Dec 02 Python
Python代码生成视频的缩略图的实例讲解
Dec 22 Python
对tensorflow中的strides参数使用详解
Jan 04 Python
Python pip配置国内源的方法
Feb 14 Python
python变量的作用域是什么
May 26 Python
Python实现打包成库供别的模块调用
Jul 13 Python
Python四款GUI图形界面库介绍
Jun 05 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
在字符串中把网址改成超级链接
2006/10/09 PHP
file_get_contents("php://input", "r")实例介绍
2013/07/01 PHP
php时间戳格式化显示友好的时间函数分享
2014/10/21 PHP
Laravel中使用自己编写类库的3种方法
2015/02/10 PHP
详解在PHP的Yii框架中使用行为Behaviors的方法
2016/03/18 PHP
PHP中file_exists使用中遇到的问题小结
2016/04/05 PHP
yii2利用自带UploadedFile实现上传图片的示例
2017/02/16 PHP
ThinkPHP3.2框架自定义配置和加载用法示例
2018/06/14 PHP
关于laravel 数据库迁移中integer类型是无法指定长度的问题
2019/10/09 PHP
js的with语句使用方法
2007/09/21 Javascript
js操作时间(年-月-日 时-分-秒 星期几)
2010/06/20 Javascript
JS实现隔行换色的表格排序
2017/03/27 Javascript
AngularJS 最常用的八种功能(基础知识)
2017/06/26 Javascript
用JS实现简单的登录验证功能
2017/07/28 Javascript
老生常谈js数据类型
2017/08/03 Javascript
Vue.js实现双向数据绑定方法(表单自动赋值、表单自动取值)
2018/08/27 Javascript
在webstorm开发微信小程序之使用阿里自定义字体图标的方法
2018/11/15 Javascript
jQuery使用ajax传递json对象到服务端及contentType的用法示例
2020/03/12 jQuery
Javascript异步执行不按顺序解决方案
2020/04/30 Javascript
JavaScript 接口原理与用法实例详解
2020/05/12 Javascript
vue实现折线图 可按时间查询
2020/08/21 Javascript
[01:02:48]2018DOTA2亚洲邀请赛小组赛 A组加赛 Newbee vs Liquid
2018/04/03 DOTA
Tensorflow设置显存自适应,显存比例的操作
2020/02/03 Python
Python日志syslog使用原理详解
2020/02/18 Python
canvas绘制圆角头像的实现方法
2019/01/17 HTML / CSS
参观接待方案
2014/03/17 职场文书
社区工作者演讲稿
2014/05/23 职场文书
小学生我的梦想演讲稿
2014/08/21 职场文书
甜品蛋糕店创业计划书
2014/09/21 职场文书
2015年春训学习心得体会范文
2015/03/09 职场文书
张思德观后感
2015/06/09 职场文书
《富饶的西沙群岛》教学反思
2016/02/16 职场文书
开学第一周日记(三篇范文)
2019/08/23 职场文书
Go语言-为什么返回值为接口类型,却返回结构体
2021/04/24 Golang
JS不要再到处使用绝对等于运算符了
2021/04/30 Javascript
python中对列表的删除和添加方法详解
2022/02/24 Python