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 18 Python
浅谈MySQL中的触发器
May 05 Python
Python中datetime常用时间处理方法
Jun 15 Python
Python的Flask框架的简介和安装方法
Nov 13 Python
简单谈谈Python中的闭包
Nov 30 Python
tensorflow识别自己手写数字
Mar 14 Python
python 除法保留两位小数点的方法
Jul 16 Python
Python wxPython库Core组件BoxSizer用法示例
Sep 03 Python
Python get获取页面cookie代码实例
Sep 12 Python
python实现自动获取IP并发送到邮箱
Dec 26 Python
django做form表单的数据验证过程详解
Jul 26 Python
Python基于jieba, wordcloud库生成中文词云
May 13 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
从C/C++迁移到PHP——判断字符类型的函数
2006/10/09 PHP
让Nginx支持ThinkPHP的URL重写和PATHINFO的方法分享
2011/08/08 PHP
php实现httpclient类示例
2014/04/08 PHP
php实现递归的三种基本方式
2020/07/04 PHP
php对xml文件的增删改查操作实现方法分析
2017/05/19 PHP
php过滤htmlspecialchars() 函数实现把预定义的字符转换为 HTML 实体用法分析
2019/06/25 PHP
jQuery1.6 使用方法二
2011/11/23 Javascript
jquery zTree异步加载简单实例分享
2013/02/05 Javascript
JS中FRAME的操作问题实例分析
2014/10/21 Javascript
JavaScript window.location对象
2014/11/14 Javascript
浅谈javascript 函数表达式和函数声明的区别
2016/01/05 Javascript
yarn与npm的命令行小结
2016/10/20 Javascript
JavaScript简单拖拽效果(1)
2017/05/17 Javascript
基于axios封装fetch方法及调用实例
2018/02/05 Javascript
在element-ui的select下拉框加上滚动加载
2019/04/18 Javascript
es6函数中的作用域实例分析
2020/04/18 Javascript
Python中获取网页状态码的两个方法
2014/11/03 Python
python实现在pandas.DataFrame添加一行
2018/04/04 Python
python3.6利用pyinstall打包py为exe的操作实例
2018/10/31 Python
Python+opencv 实现图片文字的分割的方法示例
2019/07/04 Python
python爬虫 爬取58同城上所有城市的租房信息详解
2019/07/30 Python
tensorflow指定GPU与动态分配GPU memory设置
2020/02/03 Python
使用Pycharm分段执行代码
2020/04/15 Python
世界上最大的二手相机店:KEN
2017/05/17 全球购物
数据库设计的包括哪两种,请分别进行说明
2016/07/15 面试题
中层干部竞争上岗演讲稿
2014/01/13 职场文书
公司端午节活动方案
2014/02/04 职场文书
献爱心倡议书
2014/04/14 职场文书
中药学专业毕业生推荐信
2014/07/10 职场文书
厨师长岗位职责范本
2014/08/25 职场文书
法院干警四风问题个人对照检查材料思想汇报
2014/10/07 职场文书
2014年学生会生活部工作总结
2014/11/07 职场文书
护士自荐信怎么写
2015/03/06 职场文书
《搭石》教学反思
2016/02/18 职场文书
Python制作动态字符画的源码
2021/08/04 Python
JavaScript实现外溢动态爱心的效果的示例代码
2022/03/21 Javascript