wgan.py 文件源码

def __call__(self, x, train=True):
        h = add_noise(x, test=not train)
        h = F.leaky_relu(add_noise(self.c0_0(h), test=not train))
        h = F.leaky_relu(add_noise(self.bn0_1(self.c0_1(h), test=not train), test=not train))
        h = F.leaky_relu(add_noise(self.bn1_1(self.c1_1(h), test=not train), test=not train))
        h = F.leaky_relu(add_noise(self.bn2_1(self.c2_1(h), test=not train), test=not train))
        h = F.leaky_relu(add_noise(self.bn3_0(self.c3_0(h), test=not train), test=not train))
        h = self.l4(h)
        return F.sum(h) / h.size