#include <unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h>
Converts to Tensor<Tuple<Index, Scalar> > and reduces to Tensor<Index>.