[일반] 파이토치서 nn.module 상속하면 순전파 __call__처럼 쓰이네?

[익명(218.234)]

신기하네 어떻게 만든거지;;

[해당 댓글은 삭제되었습니다.]

해당 댓글은 삭제되었습니다.

[익명(49.172)]

from typing import Any,Callable class test(): ----def forward(self): --------print(5) ----__call__ : Callable[..., Any] = forward layer = test() layer() 대충 이런식으로 구현되 있네

[익명(49.172)]

구현 보려면 여기 참조하셈 ->
https://github.com/pytorch/pytorch/blob/master/torch/nn/modules/module.py#L1156

[익명(218.235)]

오 forward를 대입시켜서 call로 사용 가능한거구나 ㄳㄳ 개신기하당