基于PyTorch的生成对抗网络实战（6）——利用Pytorch搭建WGAN-GP生成手写数字

〇、WGAN-GP介绍

WGAN-GP (Wasserstein GAN using gradient penalty)，即在WGAN的基础上加入gradient penalty梯度惩罚项，可参考论文：《Improved Training of Wasserstein GANs》。

一、生成器

生成器类与WGAN相同，代码如下：

class Generator(nn.Module):
    def __init__