python - 2次元行列を最大プールし、最大項目のインデックスも取得する最も効率的な方法は何ですか?

Question

skimage の「numpy」と「block_reduce」を使用した最大プーリングの実装に関する以下の回答を見つけました。ただし、最大値のインデックスを取得する方法は示していません (できれば、プールされたマトリックスではなく、実際のマトリックスに関して)。これに対する最も計算的に速い解決策は何ですか?

import numpy as np
import skimage.measure

a = np.array([
      [  20,  200,   -5,   23],
      [ -13,  134,  119,  100],
      [ 120,   32,   49,   25],
      [-120,   12,    9,   23]
])
skimage.measure.block_reduce(a, (2,2), np.max)