torch-nn

PyTorch의 신경망 구성 요소 라이브러리로서 선형 계층, 레이어 정규화, 드롭아웃 등 기본 블록과 텐서 연산을 결합해 모델을 모듈 단위로 정의하고 역전파를 통해 학습이 가능하도록 한다.