Tensorflow Average Height