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

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

In [1]:

# 패키지 불러오기
import torch

In [2]:

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

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

In [3]:

# 어레이를 텐서로 변경
x_tensor = torch.from_numpy(x_array)
print(x_tensor, type(x_tensor))

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

In [4]:

# 텐서를 다시 넘파이 어레이로 변환
x_array = x_tensor.numpy()
print(x_array, type(x_array))

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