How to transform Variable into numpy?

cumttang · January 21, 2017, 9:36am

When I used pytorch, I met a problem that I can not transform Variable into numpy. When I try to use torch.Tensor or transform it into torch.FloatTensor, but it’s fail. So how can I to solve this problem?

edgarriba · January 21, 2017, 9:55am

Hi, having the same issue here
https://github.com/pytorch/examples/pull/47/files#diff-1357f5fd365db925427f5f0692df5b52R251

apaszke · January 21, 2017, 12:35pm

Variable's can’t be transformed to numpy, because they’re wrappers around tensors that save the operation history, and numpy doesn’t have such objects. You can retrieve a tensor held by the Variable, using the .data attribute. Then, this should work: var.data.numpy().

cumttang · January 23, 2017, 1:17pm

Thanks a lot.

Morpheus_Hsieh · February 22, 2017, 2:30am

Hi, when I want to convert the data in a Variable x with type torch.cuda.FloatTensor into a numpy array. Use x.data.numpy() and this error pops up RuntimeError: numpy conversion for FloatTensor is not supported. Is there anything that I am missing? Thanks.

Morpheus_Hsieh · February 22, 2017, 5:38am

Ah, after more searching I found out that I gotta do cpu() first.

Robi · May 19, 2017, 3:39pm

Yes, it’s working: (Variable(x).data).cpu().numpy()

blitu12345 · December 22, 2017, 2:09pm

You can refer this link, in case converting torch tensor to numpy still poses a problem!

Royi · April 26, 2018, 8:45pm

Is there a function which handles both CUDA and CPU at once?

Nitin286roxs · March 22, 2019, 6:39am

what will type if cuda tenso is fp16 than after converting it will numpy(fp32) or numpy(fp16)?