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爬虫之爬虫的定义及URL构成
Nov 04 Python
举例讲解Python中的算数运算符的用法
May 13 Python
python 3利用Dlib 19.7实现摄像头人脸检测特征点标定
Feb 26 Python
Python 找到列表中满足某些条件的元素方法
Jun 26 Python
python绘制立方体的方法
Jul 02 Python
用python一行代码得到数组中某个元素的个数方法
Jan 28 Python
详解使用python3.7配置开发钉钉群自定义机器人(2020年新版攻略)
Apr 01 Python
关于keras.layers.Conv1D的kernel_size参数使用介绍
May 22 Python
在keras 中获取张量 tensor 的维度大小实例
Jun 10 Python
Python Opencv轮廓常用操作代码实例解析
Sep 01 Python
写一个Python脚本自动爬取Bilibili小视频
Apr 24 Python
Python基本的内置数据类型及使用方法
Apr 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
PHP5 安装方法
2006/10/09 PHP
PHP查询数据库中满足条件的记录条数(两种实现方法)
2013/01/29 PHP
ThinkPHP后台首页index使用frameset时的注意事项分析
2014/08/22 PHP
smarty实现多级分类的方法
2014/12/05 PHP
PHP匿名函数和use子句用法实例
2016/03/16 PHP
slice函数的用法 之不错的应用
2006/12/29 Javascript
不同浏览器对回车提交表单的处理办法
2010/02/13 Javascript
javascript中使用正则计算中文长度的例子
2014/04/29 Javascript
Javascript与jQuery方法的隐藏与显示
2015/01/19 Javascript
javascript中数组的定义及使用实例
2015/01/21 Javascript
基于JavaScript实现通用tab选项卡(通用性强)
2016/01/07 Javascript
JavaScript修改作用域外变量的方法
2016/03/25 Javascript
Angularjs 自定义服务的三种方式(推荐)
2016/08/02 Javascript
JavaScript实现图片瀑布流和底部刷新
2017/01/02 Javascript
微信小程序学习(4)-系统配置app.json详解
2017/01/12 Javascript
boostrapTable的refresh和refreshOptions区别浅析
2017/01/22 Javascript
js 博客内容进度插件详解
2017/02/19 Javascript
Js利用prototype自定义数组方法示例
2017/10/20 Javascript
jQuery实现右侧抽屉式在线客服功能
2017/12/25 jQuery
基于js中的存储键值对以及注意事项介绍
2018/03/30 Javascript
JavaScript设计模式之代理模式实例分析
2019/01/16 Javascript
layui时间控件选择时间范围的实现方法
2019/09/28 Javascript
JavaScript中的相等操作符使用详解
2019/12/21 Javascript
Vue+Bootstrap收藏(点赞)功能逻辑与具体实现
2020/10/22 Javascript
element-ui 弹窗组件封装的步骤
2021/01/22 Javascript
[01:33]PWL开团时刻DAY2-开雾与反开雾
2020/10/31 DOTA
Python实现多线程的两种方式分析
2018/08/29 Python
Python3爬虫中Ajax的用法
2020/07/10 Python
如何解决flask修改静态资源后缓存文件不能及时更改问题
2020/08/02 Python
极度干燥澳大利亚官方网站:Superdry澳大利亚
2019/03/28 全球购物
酒店销售经理岗位职责
2014/01/31 职场文书
腾讯广告词
2014/03/19 职场文书
公休请假条
2014/04/11 职场文书
音乐兴趣小组活动总结
2014/07/07 职场文书
合同意向书范本
2014/07/30 职场文书
仅仅使用 HTML/CSS 实现各类进度条的方式汇总
2021/11/11 HTML / CSS