本文详细介绍了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
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
