tf.estimator.Head