Answers for "numpy 3 dimensional tensor to 1d array"

1

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

Python Answers by Framework

Browse Popular Code Answers by Language