将 PyTorch CUDA 张量转换为 Z3B7F949B2343F9E5390​​E29F6EF5E1778Z 数组

Question

How do I convert a torch.Tensor (on GPU) to a numpy.ndarray (on CPU)?

Answer 1

I believe you also have to use .detach() . I had to convert my Tensor to a numpy array on Colab which uses CUDA and GPU. I did it like the following:

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() 

Answer 2

If the tensor is on gpu or cuda as you say.

You can use self.tensor.weight.data.cpu().numpy() It will copy the tensor to cpu and convert it to numpy array.

If the tensor is on cpu already you can do self.tensor.weight.data.numpy() as you correctly figured out, but you can also do self.tensor.weight.data.cpu().numpy() in this case since tensor is already on cpu , .cpu() operation will have no effect. and this could be used as a device-agnostic way to convert the tensor to numpy array.

Answer 3

some_tensor.detach().cpu().numpy()

---

• .detach() detaches from the backward graph to avoid copying gradients.
• .cpu() moves the data to CPU.
• .numpy() converts the torch.Tensor to a np.ndarray .