Community

pytorch 코드 질문이요!

안녕하세요 파이토치를 공부하고 있는 대학생입니다 파이토치 코드를 공부하다보면 m = nn.Linear(20, 30) input = torch.randn(128, 20) output = m(input) print(output.size()) torch.Size([128, 30]) 이런식으로 코드를 작성하는데, m은 Linear 클래스로 만든 인스턴스인데 아래에서 m(input)처럼 마치 함수처럼 사용하는 것이 궁금합니다.. 파이썬 callable과 관련이 있는 것 같은데 nn.Module 클래스르 상속 받아서 가능한 것인가요? 파이토치 문서를 봐도 이해가 되지 않아 질문드립니다

Loading...

알림

알림이 없습니다