这里:numpy 和 tensor数据相互转换的方法:
1.numpy->tensor
data = torch.tensor(data)
2.tensor->numpy
非训练数据(训练结束后的tensor)的转换:
data1 = np.array(data1)
如果是训练过程中需要转换,则:
data_ = data.cpu().data.numpy()
https://blog.csdn.net/qq_41368074/article/details/105560274