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脚本设置超时机制系统时间的方法
Feb 21 Python
PyCharm在win10的64位系统安装实例
Nov 26 Python
Python有序查找算法之二分法实例分析
Dec 11 Python
python实现爬山算法的思路详解
Apr 09 Python
Python面向对象思想与应用入门教程【类与对象】
Apr 12 Python
超简单的Python HTTP服务
Jul 22 Python
解决Django中修改js css文件但浏览器无法及时与之改变的问题
Aug 31 Python
TensorBoard 计算图的可视化实现
Feb 15 Python
opencv python在视屏上截图功能的实现
Mar 05 Python
python实现将字符串中的数字提取出来然后求和
Apr 02 Python
selenium框架中driver.close()和driver.quit()关闭浏览器
Dec 08 Python
python 实现Harris角点检测算法
Dec 11 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使用正则表达式提取字符串中尖括号、小括号、中括号、大括号中的字符串
2020/04/05 PHP
PHP连接MSSQL方法汇总
2016/02/05 PHP
PHP MVC框架skymvc支持多文件上传
2016/05/26 PHP
PHP单例模式模拟Java Bean实现方法示例
2018/12/07 PHP
Zero Clipboard js+swf实现的复制功能使用方法
2010/03/07 Javascript
JS中的异常处理方法分享
2013/12/22 Javascript
js格式化时间和js格式化时间戳示例
2014/02/10 Javascript
javascript中substring()、substr()、slice()的区别
2015/08/30 Javascript
JS实现3D图片旋转展示效果代码
2015/09/22 Javascript
Adapter适配器模式在JavaScript设计模式编程中的运用分析
2016/05/18 Javascript
Bootstrap fileinput组件封装及使用详解
2017/03/10 Javascript
详解使用Vue Router导航钩子与Vuex来实现后退状态保存
2017/09/11 Javascript
vue在手机中通过本机IP地址访问webApp的方法
2018/08/15 Javascript
webpack4手动搭建Vue开发环境实现todoList项目的方法
2019/05/16 Javascript
js实现抽奖功能
2020/11/24 Javascript
JavaScript canvas实现跟随鼠标移动小球
2021/02/09 Javascript
Python 深入理解yield
2008/09/06 Python
zbar解码二维码和条形码示例
2014/02/07 Python
Python3.x中自定义比较函数
2015/04/24 Python
python如何读写csv数据
2018/03/21 Python
在Pycharm中执行scrapy命令的方法
2019/01/16 Python
Python 余弦相似度与皮尔逊相关系数 计算实例
2019/12/23 Python
Python3.9最新版下载与安装图文教程详解(Windows系统为例)
2020/11/28 Python
可能这些是你想要的H5软键盘兼容方案(小结)
2019/04/23 HTML / CSS
英国最红的高街时尚品牌:Topshop
2016/08/05 全球购物
Kangol帽子官网:坎戈尔袋鼠
2018/09/26 全球购物
Schecker荷兰:狗狗用品和配件
2019/06/06 全球购物
澳洲网红粉泥面膜:Sand & Sky
2019/08/13 全球购物
常见的软件开发流程有哪些
2015/11/14 面试题
考博专家推荐信模板
2013/12/02 职场文书
八一建军节感言
2014/02/28 职场文书
学校庆元旦歌咏比赛主持词
2014/03/18 职场文书
志愿者事迹材料
2014/12/26 职场文书
2015年乡镇卫生院妇幼保健工作总结
2015/05/19 职场文书
小学英语课教学反思
2016/02/15 职场文书
mysql事务隔离级别详情
2021/10/24 MySQL