model.py 文件源码

def __call__(self, x, test=False):
        h = F.leaky_relu(self.c0(x))

        for idx in range(1, self.n_layers):
            h = F.leaky_relu(self['b{}'.format(idx)](self['c{}'.format(idx)](h), test=test))

        h = F.leaky_relu(self['b{}'.format(self.n_layers)](self['c{}'.format(self.n_layers)](h), test=test))
        h = F.sigmoid(self.c(h))

        return h