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
投稿猜你喜欢
- 今天来讲一下一些实现html中表格隔行换色的方法，即每隔一个行表格的背景色就不同，有静态html/css实现的，也有asp动态实现的。这个功
- 函数重载的替代方法-伪重载，下面看一个具体的实例代码。<? php//函数重载的替代方法-伪重载////确实，在PHP中没有函数重载这
- 函数javascript函数相信大家都写过不少了,所以我们这里只是简单介绍一下.创建函数:function f(x) {........}v
- The Only Thing We Have To Fear Is Premature Standardization原文地址：http:/
- 很多人在群里问，这个下拉框定位不到、那个弹出框定位不到…各种定位不到，其实大多数情况下就是两种问题：1 有frame，2 没有加等待。殊不知
- 我们这里所说的head区域，是指页页html代码的<head>和</head>之间的内容。在以前的文章中，主要介绍了
对象存储（Cloud Object Storage，COS）是一种存储海量文件的分布式存储服务，具有高扩展性、低成本、可靠安全等优点。对象存- 质数又称素数。指在一个大于1的自然数中，除了1和此整数自身外，不能被其他自然数整除的数。素数在数论中有着很重要的地位。比1大但不是素数的数称
- python pycharm中使用opencv时，没有代码自动补全提示解决方案有2种今天工作时突然发现，在写OPENCV相关代码时，没有自动
- 做沙盒的时候遇到一个小问题——在IE9里面竟然抓不到事件的keyCode：element.addEventListener('key
- python语句mode = ‘test’ if y is None else &lsquo
- 目的：把question_id 对应的user_answer转成ABCDsolutiondfa=df.groupby('questi
- 工作中遇到一个问题，两个字符串匹配，要求：每个字符串中最多含有一个*，?可以无限多个*代表一个任意长度的字符串，而?则代表一个字符要求可以提
- 内容摘要：一般情况下，如果我们想了解股市行情，必须登录专业股票网站，输入公司代码才能看到具体股价。其实，利用XMLHTTP协议，我们每个人都
- 下面是一个实战项目的结果。 #coding: utf-8 import Image,ImageDraw,ImageFont,os,strin
- 导语随着网购的广泛普及，现在大部分年轻人都喜欢上了网购的方式。很多东西物美价廉，出不出户也能满足你的购买需求！尤其是中秋来临，哪些假期短回不
- 在Jupyter Notebook上使用Python+opencv实现如下简单车牌字符切割。关于opencv库的安装可以参考：Python下
- 目录时间戳相减装饰器timeit模块重复调用 timeit()cProfile性能分析工具时间戳相减在代码执行前后各记录一个时间点，两个时间
- 前言这是俺写的第一篇关于python的博客，分享一下我所整合的python中的随机数函数，就当作是自己的一个笔记，也可以供像我一样的pyth
- MaxDB是MySQL AB公司通过SAP认证的数据库。MaxDB数据库服务器补充了MySQL AB产品系列。某些MaxDB特性在MySQL
