Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

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()
 
PREVIOUS NEXT
Tagged: #turn #tensor #numpy #array
ADD COMMENT
Topic
Name
8+3 =