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