np.argmax()

        np.argmax()表示返回索引最大值，需要区分索引和元素值。

        先说结论：三维张量时，将张量看成正方体，axis=0时表示沿高度轴（深度方向）各元素最大值，返回结果形状与正方体顶面相同；axis=1时表示沿宽度轴（矩阵行方向）各元素最大值，返回结果与正方体前面形状相同；axis=2时表示沿长度轴（矩阵列方向）各元素最大值，返回结果与正方体侧面形状相同。

第一维矩阵：

第二维矩阵：

 第三维矩阵：

组成的三维张量正方体：

        求b=np.argmax(a,axis=0)时，从上往下比较元素值（高度方向），取最大值，容易看到最大值将均出现在正方体底面，如max(7, 16, 25)=25，即底面，这一面各元素索引均为2，故结果为各元素均为2的3×3的矩阵。

         求c=np.argmax(a,axis=1)时，从后往前比较元素值（宽度方向），取最大值，容易看到最大值将均出现在正方体前面，如max(1, 4, 7)=7，即前面，这一面各元素索引仍均为2，故结果仍为各元素均为2的3×3的矩阵。

         求d=np.argmax(a,axis=2)时，从左往右比较元素值（长度方向），取最大值，容易看到最大值将均出现在正方体侧面，如max(7, 8, 9)=9，即侧面，这一面各元素索引仍均为2，故结果仍为各元素均为2的3×3的矩阵。

        代码结果如下。

import numpy as np

a=np.array([[[1,2,3],
             [4,5,6],
             [7,8,9]],
            [[10,11,12],
             [13,14,15],
             [16,17,18]],
            [[19,20,21],
             [22,23,24],
             [25,26,27]]])
b=np.argmax(a,axis=0)
print(b)
c=np.argmax(a,axis=1)
print(c)
d=np.argmax(a,axis=2)
print(d)
#结果
[[2 2 2]
 [2 2 2]
 [2 2 2]]

[[2 2 2]
 [2 2 2]
 [2 2 2]]

[[2 2 2]
 [2 2 2]
 [2 2 2]]