如何获得最后两个轴的argmax？

Question

假设我有一个数组，其尺寸为10 X 20 X 12 X 12 X 2 X 2。如果要获取最后两个轴的最大值索引，该怎么办？意思是我想要一个将输出（如果所描述的数组为输入）形状为10 X 20 X 12 X 12的数组的函数。因此（数组是描述的数组）：

array[indices] == np.amax(array, (-2,-1))

Answer 1

一种简单的解决方案是两次调用该函数，每个轴调用一次：

A = np.arange(1000).reshape((10,10,10))
np.amax(np.amax(A, axis=-1), axis=-1)

对于np.argmax，您基本上可以执行相同操作。但是当然，指数正在变化。您需要对此进行更正。