main.py 文件源码

def sample_manifold2d(vae, N):
    image = np.zeros((N*28, N*28))
    for z1 in xrange(N):
        for z2 in xrange(N):
            z = [np.array([norm.ppf(z1*(1/N) + 1/(2*N)),
                           norm.ppf(z2*(1/N) + 1/(2*N))])]
            sample = vae.sample(z).reshape((28, 28))
            image[z1*28:(z1 + 1)*28,z2*28:(z2 + 1)*28] = sample

    image *= 255.
    plt.imshow(image.astype(np.uint8), cmap="gray")
    plt.show()