images = tf.concat([tf.expand_dims(i, axis=0) for i in images], axis=0)