예제로 배우는 파이토치(PYTORCH)

PyTorch에는 2가지 주요한 특징

• NumPy와 유사하지만 GPU 상에서 실행 가능한 N차원 Tensor

• 신경망을 구성하고 학습하는 과정에서의 자동 미분

Tensors

준비운동 : NumPy

먼저, NumPy를 사용하여 신경망을 구성해보기.

 

PyTorch : Tensors

NumPy는 훌륭한 프레임워크지만, GPU를 사용하여 수치 연산을 가속화할 수는 없

 

 

Autograd

PyTorch : Tensor와 autograd

 

PyTorch : 새 autograd 함수 정의하기

 

 

TensorFlow : 정적 그래프(Static Graph)

 

 

nn 모듈

PyTorch : nn

 

 

PyTorch : optim

 

 

PyTorch : 사용자 정의 nn.Module

 

 

PyTorch : 제어프름(Control Flow) + 가중치 공유(Weight Sharing)