相关疑难解决方法(0)

我想使用 cuda 将 Pytorch 张量转换为 numpy 数组：

这是不使用 cuda 时的代码行：

A = self.tensor.weight.data.numpy()

如何使用 cuda 执行相同的操作？根据这个：https : //discuss.pytorch.org/t/how-to-transform-variable-into-numpy/104/3 似乎：

A = self.tensor.weight.data.cpu().numpy()