gan怎么输入一维数据_pytorch:实现简单的GAN示例(MNIST数据集)

我就废话不多说了，直接上代码吧！

# -*- coding: utf-8 -*-

"""

Created on Sat Oct 13 10:22:45 2018

@author: www

"""

import torch

from torch import nn

from torch.autograd import Variable

import torchvision.transforms as tfs

from torch.utils.data import DataLoader, sampler

from torchvision.datasets import MNIST

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.gridspec as gridspec

plt.rcParams['figure.figsize'] = (10.0, 8.0) # 设置画图的尺寸

plt.rcParams['image.interpolation'] = 'nearest'

plt.rcParams['image.cmap'] = 'gray'

def show_images(images): # 定义画图工具

images = np.reshape(images, [images.shape[0], -1])

sqrtn = int(np.ceil(np.sqrt(images.shape[0])))

sqrtimg = int(np