Question

我想使用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（）

Answer 1

如您所说，如果张量在gpu或cuda上。

您可以使用self.tensor.weight.data.cpu().numpy()，它将张量复制到cpu并将其转换为numpy数组。

如果张量已经在cpu上，则可以按照正确的方法进行self.tensor.weight.data.numpy()，但是在这种情况下，由于张量已经在{{1}上，因此您也可以执行self.tensor.weight.data.cpu().numpy() }，cpu操作将无效。这可以用作将张量转换为numpy数组的与设备无关的方法。

Answer 2

我相信您还必须使用 .detach（）。我必须在使用CUDA和GPU的Colab上将Tensor转换为numpy数组。我这样做如下：

embedding = learn.model.u_weight

embedding_list = list(range(0, 64382))

input = torch.cuda.LongTensor(embedding_list)
tensor_array = embedding(input)
# the output of the line bwlow is a numpy array
tensor_array.cpu().detach().numpy()

使用Cuda将Pytorch张量转换为Numpy数组

2 个答案: