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 14 Python
查找python项目依赖并生成requirements.txt的方法
Jul 10 Python
python实现求两个字符串的最长公共子串方法
Jul 20 Python
python程序封装为win32服务的方法
Mar 07 Python
树莓派动作捕捉抓拍存储图像脚本
Jun 22 Python
使用Python实现跳一跳自动跳跃功能
Jul 10 Python
Python学习笔记之文件的读写操作实例分析
Aug 07 Python
python自动化工具之pywinauto实例详解
Aug 26 Python
vue学习笔记之动态组件和v-once指令简单示例
Feb 29 Python
python基础详解之if循环语句
Apr 24 Python
python编程学习使用管道Pipe编写优化代码
Nov 20 Python
Python find()、rfind()方法及作用
Dec 24 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 编程的 5个良好习惯
2009/02/20 PHP
php switch语句多个值匹配同一代码块应用示例
2014/07/29 PHP
PHP内存缓存Memcached类实例
2014/12/08 PHP
php实现图片添加描边字和马赛克的方法
2014/12/10 PHP
php字符串按照单词进行反转的方法
2015/03/14 PHP
php记录搜索引擎爬行记录的实现代码
2018/03/02 PHP
PHP实现的CURL非阻塞调用类
2018/07/26 PHP
jQuery 打造动态下滑菜单实现说明
2010/04/15 Javascript
JavaScript入门之事件、cookie、定时等
2011/10/21 Javascript
一张表格告诉你windows.onload()与$(document).ready()的区别
2014/05/16 Javascript
jQuery简单操作cookie的插件实例
2016/01/13 Javascript
jquery中键盘事件小结
2016/02/24 Javascript
node.js使用cluster实现多进程
2016/03/17 Javascript
javascript使用 concat 方法对数组进行合并的方法
2016/09/08 Javascript
jQuery弹出窗口简单实现代码
2017/03/09 Javascript
vue父组件通过props如何向子组件传递方法详解
2017/08/16 Javascript
vue 循环加载数据并获取第一条记录的方法
2018/09/26 Javascript
React SSR样式及SEO的实践
2018/10/22 Javascript
angular6 利用 ngContentOutlet 实现组件位置交换(重排)
2018/11/02 Javascript
使用python 获取进程pid号的方法
2014/03/10 Python
python实现爬虫统计学校BBS男女比例(一)
2015/12/31 Python
python字符串,数值计算
2016/10/05 Python
基于pytorch的保存和加载模型参数的方法
2019/08/17 Python
Python3读写Excel文件(使用xlrd,xlsxwriter,openpyxl3种方式读写实例与优劣)
2020/02/13 Python
pandas处理csv文件的方法步骤
2020/10/16 Python
贝玲妃美国官方网站:Benefit美国
2016/08/28 全球购物
当当网官方旗舰店:中国图书销售夺金品牌
2018/04/02 全球购物
屈臣氏越南官网:Watsons越南
2021/01/14 全球购物
八年级音乐教学反思
2014/01/09 职场文书
分家协议书
2014/04/21 职场文书
工地安全质量标语
2014/06/07 职场文书
社区党建工作方案
2014/06/10 职场文书
劳模先进事迹材料
2014/12/24 职场文书
会计简历自我评价
2015/03/10 职场文书
2016学习雷锋精神活动倡议书
2015/04/27 职场文书
恰同学少年观后感
2015/06/08 职场文书