【pytorch】數據類型

Torch 定義了九種CPU tensor類型和九種GPU tensor類型

Data type	dtype	CPU tensor	GPU tensor
32位浮點型	torch.float32 或 torch.float	torch.FloatTensor	torch.cuda.FloatTensor
64位浮點型	torch.float64 或 torch.double	torch.DoubleTensor	torch.cuda.DoubleTensor
16位浮點型	torch.float16 或 torch.half	torch.HalfTensor	torch.cuda.HalfTensor
8位整型 (無符號)	torch.uint8	torch.ByteTensor	torch.cuda.ByteTensor
8位整型 (有符號)	torch.int8	torch.CharTensor	torch.cuda.CharTensor
16位整型 (有符號)	torch.int16 或 torch.short	torch.ShortTensor	torch.cuda.ShortTensor
32位整型 (有符號)	torch.int32 或 torch.int	torch.IntTensor	torch.cuda.IntTensor
64位整型 (有符號)	torch.int64 或 torch.long	torch.LongTensor	torch.cuda.LongTensor
布爾型	torch.bool	torch.BoolTensor	torch.cuda.BoolTensor

torch.Tensor 是默認的tensor類型 (torch.FloatTensor)的別名。