(pytorch) pytorch-tensor와 numpy-ndarray 간에 변환하기

(1) ndarray -> tensor

a = np.array([1,2])
t = torch.from_numpy(a)

 

(2) tensor -> ndarray

t = torch.tensor([1,2])
a = t.numpy()