[PyTorch] nn.Sequential


[PyTorch] nn.Sequential

nn.Sequential 클래스는 nn.Linear, nn.ReLU(활성화 함수) 같은 모듈들을 인수로 받아서 순서대로 정렬해놓고 입력값이 들어모면 순서대로 모듈을 실행해서 결과값을 리턴한다. 신경망의 깊이가 깊어질수록 묶어서 연산하는 것이 더 편리하기에 nn.Sequential 클래스를 사용하는 것이 좋다...

[PyTorch] nn.Sequential에 대한 요약내용입니다.

자세한 내용은 아래에 원문링크를 확인해주시기 바랍니다.



원문링크 : [PyTorch] nn.Sequential