📜  无法将 CUDA 张量转换为 numpy.使用 Tensor.cpu() 复制张量 - 任何代码示例

📅  最后修改于: 2022-03-11 14:55:36.132000             🧑  作者: Mango

代码示例1
# 在我們想把 GPU tensor 轉換成 Numpy 時,需要先將 tensor 轉換到 CPU 去,
# 因為 Numpy 是 CPU-only 的。

# images.numpy() => images.cpu().numpy()