import tensorflow as tf

    # For printing layers shape
    self.training_end_points = self.end_points_D
    self.training_end_points.update(self.end_points_G)

    tf.summary.histogram("d", self.end_points_D['D_on_data'])
    tf.summary.histogram("d_", self.end_points_D['D_on_G'])
    tf.summary.image("G", G)

