Add support for any rank ArgMin. Hacky workaround for IndexType tensors by doing type-conversion after operation