PyTorch与NumPy数据类型转换完全指南

文章目录

• • • tensor to numpy
      • numpy to tensor

tensor to numpy

tensor数据在cpu上
：

如果tensor数据在cpu上，直接使用
.numpy()
就可以转化。

例子：

注意：torch 和 numpy 转化后 指向地址相同

如果修改原始数据，那么转换后的数据也会修改，例子：

tensor数据在gpu上
：

如果tensor数据在gpu上，那么需要将tensor数据先转移到cpu上面，然后在进行转化。

例子

a = torch.randn(2,3,4)
a = a.cuda()
print('type of a ', type(a))
print('device of a', a.device)

b = a.numpy() # 会出错

报