VAE变分自编码器

变分自编码器

以mnist数据集为例

载入数据

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('MNIST_data')

定义输入数据和输出数据

encoder的输入是一个batch的图片。shape为 [None，28,28] ,单色通道。输出为根据该图片产生的专属于每一个$x_k$的$m_{}$