Pytorch中Tensor的类型转换

Pytorch中的Tensor常用的类型转换函数（inplace操作）：

（1）数据类型转换

在Tensor后加 .long(), .int(), .float(), .double()等即可，也可以用.to()函数进行转换，所有的Tensor类型可参考
　　 https://pytorch.org/docs/stable/tensors.html

（2）数据存储位置转换

CPU张量 ----> GPU张量，使用data.cuda()

GPU张量 ----> CPU张量，使用data.cpu()

（3）与numpy数据类型转换

Tensor---->Numpy 使用 data.numpy()，data为Tensor变量

Numpy ----> Tensor 使用 torch.from_numpy(data)，data为numpy变量

（4）与Python数据类型转换

Tensor ----> 单个Python数据，使用data.item()，data为Tensor变量且只能为包含单个数据

Tensor ----> Python list，使用data.tolist()，data为Tensor变量，返回shape相同的可嵌套的list

参考：pytorch官网