Answers for "np ndarray to tensor"

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

convert numpy array to tensor

import numpy as np
import torch

numpy_array = np.array([1,3,5])
tensor_array = torch.from_numpy(numpy_array)

Posted by: Guest on July-23-2021

Code answers related to "np ndarray to tensor"