变量类型

1. 基本数据类型

2. 类型转换

2.1 CPU和GPU的Tensor之间转换

从cpu –> gpu，使用data.cuda()即可。
若从gpu –> cpu，则使用data.cpu()。

2.2 Tensor与Numpy Array之间的转换

Tensor –> Numpy.ndarray 可以使用 data.numpy()，其中data的类型为torch.Tensor。
Numpy.ndarray –> Tensor 可以使用torch.from_numpy(data)，其中data的类型为numpy.ndarray。

2.3 Tensor的基本类型转换

为了方便测试，我们构建一个新的张量，你要转变成不同的类型只需要根据自己的需求选择即可

tensor = torch.Tensor(2, 5)

–> long : torch.long()
newtensor = tensor.long()

–> 半精度浮点(16位浮点)类型: torch.half()
newtensor = tensor.half()

–> int类型: torch.int()
newtensor = tensor.int()

–> double类型: torch.double()
newtensor = tensor.double()

–> float类型: torch.float()
newtensor = tensor.float()

–> char类型: torch.char()
newtensor = tensor.char()

–> byte类型: torch.byte()
newtensor = tensor.byte()

–>short类型: torch.short()
newtensor = tensor.short()