pytorch tensor类型转换

tensor转numpy

Tensor—->Numpy  使用 data.numpy()，data为Tensor变量  链接：http://ddrv.cn/a/575619

 

另：

（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相同的可嵌套的

 

 

 

 

 

 

tensor中只有一个数时，直接转换成数 链接：https://pytorch.org/docs/stable/tensors.html

>>> x = torch.tensor([[1]])
>>> x
tensor([[ 1]])
>>> x.item()
1
>>> x = torch.tensor(2.5)
>>> x
tensor(2.5000)
>>> x.item()
2.5