这里: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

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐