Tensorflow Tf Math Maximum Point