Argmax массива numpy, возвращающего неплоские индексы

Я пытаюсь получить индексы максимального элемента в массиве Numpy. Это можно сделать с помощью numpy.argmax . Моя проблема в том, что я хотел бы найти самый большой элемент во всем массиве и получить его индексы.

numpy.argmax можно применить либо вдоль одной оси, что я не хочу, либо к уплощенному массиву, что мне и нужно.

Моя проблема в том, что использование numpy.argmax с axis = None возвращает плоский индекс, когда мне нужен многомерный индекс.

Я мог бы использовать divmod , чтобы получить не плоский индекс, но это выглядит некрасиво. Есть ли лучший способ сделать это?