pytorch中tensor的类型

最新推荐文章于 2025-05-31 17:40:48 发布

Jet__P

最新推荐文章于 2025-05-31 17:40:48 发布

阅读量417

点赞数

CC 4.0 BY-SA版权

文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/Jet__P/article/details/118531711

本文详细介绍了Torch库中定义的CPU和GPU两种环境下的七种和八种tensor类型，包括不同位宽的浮点数和整数类型。CPU tensor涵盖了从FloatTensor到LongTensor的各种精度，而GPU tensor则在浮点数中增加了HalfTensor，并同样提供了从ByteTensor到LongTensor的整数类型。了解这些类型对于高效利用GPU资源进行深度学习计算至关重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Torch定义了七种CPU tensor类型和八种GPU tensor类型：

CPU tensor
32-bit floating point： torch.FloatTensor
64-bit floating point： torch.DoubleTensor
8-bit integer (unsigned)： torch.ByteTensor
8-bit integer (signed)： torch.CharTensor
16-bit integer (signed) ： torch.ShortTensor
32-bit integer (signed) ： torch.IntTensor
64-bit integer (signed) ： torch.LongTensor

GPU tensor
32-bit floating point： torch.cuda.FloatTensor
64-bit floating point： torch.cuda.DoubleTensor
16-bit floating point： torch.cuda.HalfTensor
8-bit integer (unsigned)： torch.cuda.ByteTensor
8-bit integer (signed)： torch.cuda.CharTensor
16-bit integer (signed) ： torch.cuda.ShortTensor
32-bit integer (signed) ： torch.cuda.IntTensor
64-bit integer (signed) ： torch.cuda.LongTensor

torch.Tensor是默认类型torch.FloatTensor