[ML][pytorch] torch.nn 과 torch.nn.Functional 의 차이

결론부터 말하자면

이름처럼 torch.nn.functional은 함수고 torch.nn은 클래스이다.

 

그래서 torch.nn으로 구현한 경우에는 객체를 인스턴스화하여 attribute를 활용해 state를 저장하고 활용할 수 있고,

torch.nn.functional로 구현한 경우에는 인스턴스화 시킬 필요 없이 바로 사용이 가능하다.

*attribute란 클래스 내부에 있는 메소드나 변수 등을 의미함

 

그저 model class를 만들 때 init 부분에 torch.nn 클래스를 이용하여 모델을 정의해 버리거나 forward 진행할 때 직접 torch.nn.functional 함수를 이용하여 계산해주거나의 차이일 뿐!

 

출처

https://cchhoo407.tistory.com/29

https://cvml.tistory.com/10