nn.py 文件源码

def set_submodule_layers(self, namespace, module):
        with self.init_scope():
            for index, layer in enumerate(module.__layers__):
                if isinstance(layer, chainer.Link):
                    self.super__setattr__("_nn_{}_layer_{}".format(namespace, index), layer)

                if isinstance(layer, Residual):
                    for resnet_index, _layer in enumerate(layer.__layers__):
                        if isinstance(_layer, chainer.Link):
                            self.super__setattr__("_nn_{}_layer_{}_res_{}".format(namespace, index, resnet_index), _layer)