본문으로 건너뛰기
재미있는 삽질 기록
Home
Categories
Home
Categories
#nn-module
Linear Layer와 ReLU를 직접 구현해보기
nn.Linear 와 nn.ReLU 라는 이미 완성된 클래스가 존재하지만, 직접 구현해보면 더 명확하게 이해할 수 있고 재미있는 사실들도 발견할 수 있다. torch.nn 없이 직접 구현해보았다. Linear Layer 구현하기 선형 계층은 입력에 가중치를 곱하고 편향을 더하는 y...
#pytorch #nn-module #relu
|
2025-09-05