#include <contrib/Eigen/src/Tensor/TensorArgMax.h>
Converts to Tensor<Pair<Index, Scalar> > and reduces to Tensor<Index>.