add.layer 말고도 다른 방법의 구현
*args 로 가변 인수를 받아 레이어 쌓기
pytorch 의 경우 예시
@overload
def __init__(self, *args: Module) -> None:
...
@overload
def __init__(self, arg: 'OrderedDict[str, Module]') -> None:
...
def __init__(self, *args):
super().__init__()
if len(args) == 1 and isinstance(args[0], OrderedDict):
for key, module in args[0].items():
self.add_module(key, module)
else:
for idx, module in enumerate(args):
self.add_module(str(idx), module)
Sequential 클래스의 __init__ 메서드
OrderedDict 형태와 OrderedDict 이 아닌 여러 개의 모듈들이 개별적으로 전달된 경우가 존재
모듈을 사용자가 지정한 이름으로 추가하는 것,
'dev_AI_framework' 카테고리의 다른 글
RNN 계산 그래프 구현 (3) | 2024.10.14 |
---|---|
pooling 연산의 구현 (0) | 2024.10.04 |
결국 어디에 포커스를 둘 것인가... (0) | 2024.09.30 |
pytorch 가 layer 를 쌓는 과정에 대해, activation layer 의 구현 (0) | 2024.09.30 |
pytorch 에 기반, 계산 그래프의 생성 방식을 바꿔보자... (0) | 2024.09.30 |