vae.py 文件源码

def build_graph(self):
        """
        Builds graph
        """

        # Clear the grapht
        tf.reset_default_graph()

        # Create the placeholder for the input
        nx = self.enc_dim[0]
        self.x = tf.placeholder("float", shape=[None, nx], name='x')


        # Builds the various components        
        if self.mode == 'train':
            self.build_enc()
        self.build_dec()
        self.build_loss_fn()

        # Add the summary op
        self.summary_op = tf.summary.merge_all()

        # Create a saver
        self.saver = tf.train.Saver()