pytorch 코드 질문이요!

안녕하세요! 단순히 공식 문서의 소스 코드만 봤을때는 Module 클래스를 상속 받아서 가능한것 같습니다. Module 클래스를 보시면 __call__ 이라는 내장 함수에 _call_impl라는 함수를 할당해주고 있어요. 파이썬에서 __call__은 클래스를 함수처럼 invoke 해주는 내장 함수입니다 :) - https://pytorch.org/docs/stable/_modules/torch/nn/modules/module.html#Module - https://wjunsea.tistory.com/61

1
2
3
4
5
6
7
# https://pytorch.org/docs/stable/_modules/torch/nn/modules/module.html#Module ...
class Module:
  # 생략  
  def _call_impl(self, *input, **kwargs):
    # 생략
    return result
  __call__ = _call_impl # <- 이게 있어서 m(input)으로 호출하면 __call__ 함수가 실행됩니다.