pytorch tensor to numpy code example

Example 1: tensot to numpy pytorch

#Back and forth between torch tensor and numpy

#np --> tensot
torch.from_numpy(your_numpy_array)

#tensor --> np
your_torch_tensor.numpy()

Example 2: convert pytorch tensor to numpy

na = a.to('cpu').numpy()

Example 3: convert torch to numpy

your_tensor.numpy()

Example 4: tensor.numpy() pytorch gpu

tensor_array.cpu().detach().numpy()