tensor.gpu转numpy.cpu
# pre为tensor.gpu
# numpy_pre为numpy.cpu
numpy_pre = torch.squeeze(pre).cpu().detach().numpy()
numpy.cpu转tensor.gpu
#num为numpy.cpu
#gpu_tensor为gpu.tensor
gpu_tensor = torch.from_numpy(num).cuda()
整理不易,欢迎一键三连!!!
送你们一条美丽的--分割线--
🌷🌷🍀🍀🌾🌾🍓🍓🍂🍂🙋🙋🐸🐸🙋🙋💖💖🍌🍌🔔🔔🍉🍉🍭🍭🍋🍋🍇🍇🏆🏆📸📸⛵⛵⭐⭐🍎🍎👍👍🌷🌷