Related to: Machine Learning

torch.nn.Module

  • ai 모델을 구성하는 layer의 base class
  • input, output, forward, backward를 정의해야 함
  • 학습 대상이 되는 parameter(tensor)를 정의해야 함

참조

Week 2