Answers for "turn torch tensor into numpy array"

convert pytorch tensor to numpy

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

Posted by: Guest on January-20-2021

how do i turn a tensor into a numpy array

import torch

# Create PyTorch tensor
A_torch = torch.tensor([1, 2])

# Convert tensor to NumPy array
A_np = A_torch.numpy()

Posted by: Guest on January-14-2021

Code answers related to "turn torch tensor into numpy array"