pytorch中tensor的类型

最新推荐文章于 2024-02-16 00:00:38 发布

原创最新推荐文章于 2024-02-16 00:00:38 发布 · 444 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

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

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