将 PyTorch CUDA 张量转换为 NumPy 数组

Question

如何将 torch.Tensor （在 GPU 上）转换为 numpy.ndarray （在 CPU 上）？

原文由 Noa Yehezkel 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

使用 .detach() 从 GPU / CUDA 张量转换为 numpy 数组：

 tensor.detach().cpu().numpy()

原文由 azizbro 发布，翻译遵循 CC BY-SA 4.0 许可协议