在将数据转为Tensor训练完之后,如何将张量转回数组,如下:
img.cpu().detach().numpy()
GPU后将Tensor转换为numpy
最新推荐文章于 2024-06-09 11:01:50 发布
在将数据转为Tensor训练完之后,如何将张量转回数组,如下:
img.cpu().detach().numpy()