[PyTorch] NumPy 변환 (Bridge)

NumPy 변환 (Bridge)

 

CPU 상의 tensor와 numpy array는 메모리 공간을 공유하기 때문에, 하나를 변경하면 다른 하나도 변경된다.

 

# tensor를 numpy array로 변환하기
t = torch.ones(5)
print(f"t: {t}")
n = t.numpy()
print(f"n: {n}")

Out:

t: tensor([1., 1., 1., 1., 1.])
n: [1. 1. 1. 1. 1.]

 

tensor의 변경 사항이 numpy array에 반영된다.

 

t.add_(1)
print(f"t: {t}")
print(f"n: {n}")

Out:

t: tensor([2., 2., 2., 2., 2.])
n: [2. 2. 2. 2. 2.]

 

numpy array를 tensor로 변환하기

 

n = np.ones(5)
t = torch.from_numpy(n)

 

numpy array의 변경 사항이 tensor에 반영된다.

 

np.add(n, 1, out=n)
print(f"t: {t}")
print(f"n: {n}")

Out:

t: tensor([2., 2., 2., 2., 2.], dtype=torch.float64)
n: [2. 2. 2. 2. 2.]

 

https://tutorials.pytorch.kr/beginner/basics/tensorqs_tutorial.html

텐서(Tensor) — PyTorch Tutorials 1.10.0+cu102 documentation