import tensorflow as tf


      inputs = tf.reshape(inputs, [-1, self.final_size])

      inputs = tf.layers.dense(inputs=inputs, units=self.num_classes)
      inputs = tf.identity(inputs, 'final_dense')
      return inputs
