把tensor转为numpy_torch.tensor的类型转换以及和numpy的转换

PyTorch中的常用的tensor类型

PyTorch中的常用的tensor类型包括：

32位浮点型torch.FloatTensor，

64位浮点型torch.DoubleTensor，

16位整型torch.ShortTensor，

32位整型torch.IntTensor，

64位整型torch.LongTensor。

类型之间的转换

一般只要在tensor后加long(), int(), double(),float(),byte()等函数就能将tensor进行类型转换

此外，还可以使用type()函数，data为Tensor数据类型，data.type()为给出data的类型，如果使用data.type(torch.FloatTensor)则强制转换为torch.FloatTensor类型张量。

a1.type_as(a2)可将a1转换为a2同类型。

tensor和numpy.array转换

tensor -> numpy.array: data.numpy()，如：

numpy.array -> tensor: torch.from_numpy(data)，如：

CPU张量和GPU张量之间的转换

CPU -> GPU: data.cuda()

GPU -> CPU: data.cpu()

参考资料：