pytorch常用数据类型所占字节数对照表一览
更新时间：2021年05月17日 09:05:20 投稿：jingxian
这篇文章主要介绍了pytorch常用数据类型所占字节数对照表一览，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
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绘制柱状图可视化神器pyecharts,文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-09-09
这篇文章主要介绍了如何利用Python编写一个简单的微博爬虫，感兴趣的小伙伴们可以参考一下2016-03-03
python用fsolve、leastsq对非线性方程组求解
这篇文章主要为大家详细介绍了python用fsolve、leastsq对非线性方程组进行求解，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-12-12
这篇文章主要介绍了Python 列表的索引取值,本节重点掌握多次索引取值的语法：列表[索引][索引]，结合示例代码给大家介绍的非常详细，需要的朋友可以参考下2022-09-09
这篇文章主要介绍了linux之父进程使用kill函数杀死子进程方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-06-06
Modbus TCP 是一种基于 TCP/IP 协议栈的 Modbus 通信协议,它用于在工业自动化系统中进行设备之间的通信,只要通过pymodbus或pyModbusTCP任意模块就可以实现,本文采用pymodbus,感兴趣的朋友跟随小编一起看看吧2023-10-10
手把手教你快速安装gpu版本的pytorch(详细图文教程)
在Windows 10上安装PyTorch时,通常默认安装的是CPU版本,且下载速度较慢,本文提供了一个详细的安装指南,包括如何检查CUDA版本、选择合适的PyTorch、torchvision和torchaudio版本,并通过pip而非conda进行安装,文中通过图文介绍的非常详细,需要的朋友可以参考下2024-09-09
本文主要介绍了pandas计算相关系数corr返回空的问题解决，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-01-01
这篇文章主要介绍了Python中特殊函数，主要介绍一下四个函数：1 过滤函数filter 2 映射和归并函数map/reduce 3 装饰器@ 4 匿名函数lamda,需要的朋友可以参考下2015-07-07
这篇文章主要介绍了Python的gevent框架的入门教程,示例代码基于Python2.x版本,需要的朋友可以参考下2015-04-04
最新评论