torch.FloatTensor()生成的数据类型dtype大小

最新推荐文章于 2025-09-28 10:55:14 发布

原创最新推荐文章于 2025-09-28 10:55:14 发布 · 9.8k 阅读

20 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #张量 #深度学习

该博客介绍了PyTorch中torch.FloatTensor()创建的张量数据类型，默认为32位浮点数，并提供了官网链接以查阅更多张量对应的数据类型。

部署运行你感兴趣的模型镜像

结论

torch.FloatTensor()默认生成32位浮点数，dtype 为 torch.float32 或 torch.float

`验证`

     >>> tensor = torch.FloatTensor([[1,2],[3,4]])
     >>> print tensor
             tensor([[1., 2.],
             [3., 4.]])
     >>> print tensor.dtype    
             torch.float32

官网各类tensor对应数据类型如下：

https://pytorch.org/docs/stable/tensors.html?highlight=torch%20floattensor

Data type	dtype	CPU tensor	GPU tensor
32-bit floating point	`torch.float32` or `torch.float`	`torch.FloatTensor`	`torch.cuda.FloatTensor`
64-bit floating point	`torch.float64` or `torch.double`	`torch.DoubleTensor`	`torch.cuda.DoubleTensor`
16-bit floating point 1	`torch.float16` or `torch.half`	`torch.HalfTensor`	`torch.cuda.HalfTensor`
16-bit floating point 2	`torch.bfloat16`	`torch.BFloat16Tensor`	`torch.cuda.BFloat16Tensor`
32-bit complex	`torch.complex32`
64-bit complex	`torch.complex64`
128-bit complex	`torch.complex128` or `torch.cdouble`
8-bit integer (unsigned)	`torch.uint8`	`torch.ByteTensor`	`torch.cuda.ByteTensor`
8-bit integer (signed)	`torch.int8`	`torch.CharTensor`	`torch.cuda.CharTensor`
16-bit integer (signed)	`torch.int16` or `torch.short`	`torch.ShortTensor`	`torch.cuda.ShortTensor`
32-bit integer (signed)	`torch.int32` or `torch.int`	`torch.IntTensor`	`torch.cuda.IntTensor`
64-bit integer (signed)	`torch.int64` or `torch.long`	`torch.LongTensor`	`torch.cuda.LongTensor`
Boolean	`torch.bool`	`torch.BoolTensor`	`torch.cuda.BoolTensor`
quantized 8-bit integer (unsigned)	`torch.quint8`	`torch.ByteTensor`	/
quantized 8-bit integer (signed)	`torch.qint8`	`torch.CharTensor`	/
quantized 32-bit integer (signed)	`torch.qfint32`	`torch.IntTensor`	/
quantized 4-bit integer (unsigned) 3	`torch.quint4x2`	`torch.ByteTensor`	/