models.py 文件源码

def __call__(self, x):
        h = F.leaky_relu(self.c0(x))   
        h = F.leaky_relu(self.bn1(self.c1(h)))
        h = F.leaky_relu(self.bn2(self.c2(h)))
        h = F.leaky_relu(self.bn3(self.c3(h)))
        l = self.l4l(h)
        return l