nn.py 文件源码

def __init__(self, *layers):
        super(Module, self).__init__()
        self.layers = []
        self.blocks = []
        self.links = []
        self.modules = []
        self._locked = False
        if len(layers) > 0:
            self.add(*layers)