pytorch常用数据类型所占字节数对照表一览
作者：jingxian 发布时间：2021-07-25 15:26:29
标签：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)
来源：https://blog.csdn.net/yrwang_xd/article/details/106255054
0
投稿猜你喜欢
- 在开发高并发系统时，有三把利器用来保护系统：缓存、降级和限流。那么何为限流呢？顾名思义，限流就是限制流量，就像你宽带包了1个G的流量，用完了
- 大家好，今天在写代码的时候，遇到了这样一种情况。我有如下所示的几个类用来存放程序配置（其实当做命名空间来用，同时感觉能够继承方便一点），im
- 问题某些无聊的脚本小子在Web页面表单中填入了“pýtĥöñ”这样的文本，我们
前言注释可以起到一个备注的作用，团队合作的时候，个人编写的代码经常会被多人调用，为了让别人能更容易理解代码的通途，使用注释是非常有效的。Py- 一、os模块os 模块是 Python中的一个内置模块，也是 Python中整理文件和目录最为常用的模块。该模块提供了非常丰富的方法用来处理
- 一、model.py1.1 Channel Shuffledef channel_shuffle(x: Tensor, groups: in
- DIV与TABLE本身并不存在什么优缺点，所谓web标准只是推荐的是正确的使用标签，好比说：DIV用于布局，而TABLE则本来就是转二维数据
- 1. 加载数据集这次我们搭建一个小小的多层线性网络对糖尿病的病例进行分类首先先导入需要的库文件先来看看我们的数据集观察可以发现，前八列是我们
- 前言图像分割在医学成像、自动驾驶汽车和卫星成像等方面有很多应用，本质其实就是图像像素分类任务，也就是使用深度学习模型为输入图像的每个像素分配
- 前言容器数据类型包括数组list，字典dict以及元组tuple等。本篇，将详细介绍ChainMap字典序列的使用。ChainMapChai
- 今天给大家分享一个用原生JS实现的好看计数器，效果如下：以下是代码实现，欢迎大家复制粘贴和收藏。<!DOCTYPE html>&
- 1.首先axios不支持vue.use()方式声明使用看了所有近乎相同的axios文档都没有提到这一点建议方式在main.js中如下声明使用
- ScrapyScrapy是纯python实现的一个为了爬取网站数据、提取结构性数据而编写的应用框架。Scrapy使用了Twisted异步网络
- 本文实例讲述了Python爬取需要登录的网站实现方法。分享给大家供大家参考，具体如下：import requestsfrom lxml im
- 简介这里通过python的绘图工具Matplotlib包可视化实现机器学习中的KNN算法。 需要提前安装python的Numpy和Matpl
- public bool SaveSMSMessage(SMSBatch smsBatch, DataSet smsMessages) { /
- 一、MySQL的体系结构① 结构图② 各结构的功能连接层：最上层是一些客户端和链接服务，主要完成一些类似于连接处理、授权认证、及相关的安全方
- 本文实例为大家分享了python使用Matplotlib绘制分段函数的具体代码，供大家参考，具体内容如下环境Python3Mac OS代码#
- 有一个古老的争论，是关于在哪里存储应用程序业务逻辑的：是在应用程序本身的业务逻辑层中还是在数据库层中。应用程序逻辑层的绝对支持者提出，数据库
- 本文实例讲述了JavaScript 扩展运算符用法。分享给大家供大家参考，具体如下：扩展运算符格式扩展运算符格式很简单，就是三个点(…)重点
