파이토치 어레이를 텐서 타입으로 array to tensor pytorch

파이토치(torch) 어레이를 텐서 타입으로 array to tensor pytorch¶

In [1]:

import torch

In [2]:

# 어레이 생성
import numpy as np
x = np.array([[1, 2], 
              [3, 4]])
print(x, type(x))

[[1 2]
 [3 4]] <class 'numpy.ndarray'>

In [3]:

# 어레이를 텐서로 변환 : from_numpy 사용
x = torch.from_numpy(x)
print(x, type(x))

tensor([[1, 2],
        [3, 4]]) <class 'torch.Tensor'>

In [4]:

# 텐서를 넘파이로 변환 : numpy() 사용
x = x.numpy()
print(x, type(x))

[[1 2]
 [3 4]] <class 'numpy.ndarray'>