models.py 文件源码

def __call__(self, x, test=False):
        h = F.leaky_relu(self.c0(x))
        h = F.leaky_relu(self.bn_c1(self.c1(h), test=test))
        h = F.leaky_relu(self.bn_c2(self.c2(h), test=test))
        h = self.c3(h)
        h = F.sum(h) / h.size  # Mean
        return h