Pytorch data types and type conversion

W_dyyy

已于 2024-07-10 07:18:20 修改

阅读量1k

点赞数 14

分类专栏：机器学习 DeepLearning 文章标签： pytorch 人工智能 python

于 2024-06-28 11:26:38 首次发布

本文链接：https://blog.youkuaiyun.com/qq_35455206/article/details/140034321

版权

在pytorch中，tensor是我们处理的主要对象。而在pytorch 0.4.0之后，当我们需要梯度时，我们不必再使用Variable而是直接在tensor中指定requires_grad = True。当然，如果我们不需要梯度，我们就将自己的代码放入with torch.no_grad():中。本篇文章将结合pytorch文档以及pytorch开发者smth在社区的一些回答来总结pytorch数据类型和转换的一些技巧和逻辑。

1 tensor 类型

常见的tensor datatypes
Data type	dtype
32-bit floating point	`torch.float32` or `torch.float`
64-bit floating point	`torch.float64` or `torch.double`
16-bit floating point	`torch.float16` or `torch.half`
8-bit integer (unsigned)	`torch.uint8`
8-bit integer (signed)	`torch.in`