Answers for "ndarray to tensor pytorch"

2

numpy array to torch tensor

np_array = np.array(data)
x_np = torch.from_numpy(np_array)
Posted by: Guest on July-14-2021
0

convert pytorch tensor to numpy

na = a.to('cpu').numpy()
Posted by: Guest on January-20-2021

Python Answers by Framework

Browse Popular Code Answers by Language