在PyTorch中，一些数据类型需要掌握，那么如何才能够自定义数据类型。接下来就是一些参考。
| 数据类型 | dtype | CPU张量 | GPU张量 |
|---|---|---|---|
| 32位浮点数（float) | torch.float32 or torch.float | torch.FloatTensor | torch.cuda.FloatTensor |
| 64位浮点数（double） | torch.float64 or torch.double | torch.DoubleTensor | torch.cuda.DoubleTensor |
| 16位浮点数（half） | torch.float16 or torch.half | torch.HalfTensor | torch.cuda.HalfTensor |
| 16位浮点数（bfloat16) | torch.bfloat16 | torch.BFloat16Tensor | torch.cuda.BFloat16Tensor |
| 32位复合型 | torch.complex32 | ||
| 64位复合型 | torch.complex64 | ||
| 128位复合型 | torch.complex128 or torch.cdouble | ||
| 8-bit integer (unsigned) | torch.uint8 | torch.ByteTensor | torch.cuda.ByteTensor |
| 8带符号位整型(int8) | torch.int8 | torch.CharTensor | torch.cuda.CharTensor |
| 16位带符号整型 (short) | torch.int16 or torch.short | torch.ShortTensor | torch.cuda.ShortTensor |
| 32位带符号整型 (int) | torch.int32 or torch.int | torch.IntTensor | torch.cuda.IntTensor |
| 64位带符号整型 (long) | torch.int64 or torch.long | torch.LongTensor | torch.cuda.LongTensor |
| 布尔逻辑 | torch.bool | torch.BoolTensor | torch.cuda.BoolTensor |
