mnist_dcgan.py 文件源码

def __call__(self, x):
        h = self.bn1(F.leaky_relu(self.l1(x)), test=not self.train)
        h = self.bn2(F.leaky_relu(self.l2(h)), test=not self.train)
        h = self.bn3(F.leaky_relu(self.l3(h)), test=not self.train)
        h = self.bn4(F.leaky_relu(self.l4(h)), test=not self.train)
        y = self.l5(h)
        return y