pytorch中tensor的类型

原创于 2021-07-06 22:58:42 发布 · 417 阅读

CC 4.0 BY-SA版权

文章标签：

本文详细介绍了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