더북(TheBook)

단순한 Sequential 클래스

이제 NaiveSequential 클래스를 만들어 층을 연결해 보겠습니다. 층의 리스트를 받고 __call__() 메서드에서 입력을 사용하여 층을 순서대로 호출합니다. 층의 파라미터를 쉽게 구할 수 있도록 weights 속성을 제공합니다.

class NaiveSequential:
    def __init__(self, layers):
        self.layers = layers  

    def __call__(self, inputs):
        x = inputs
        for layer in self.layers:
           x = layer(x)
        return x  
        
    @property 
    def weights(self):
       weights = []
       for layer in self.layers:
           weights += layer.weights
       return weights
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.