对于numpy中的函数的参数dim的一点理解

对于numpy中的函数的参数dim的一点理解

经常被dim参数搞混。试着总结了一下。记忆瞬间清晰了

以.max(dim)方法为例：

>>> import numpy as np
>>> a = np.random.randint(1, 100, [2, 3, 4])
>>> a
array([[[26, 36, 31, 21],
        [74, 59, 79, 32],
        [77, 94, 81, 32]],

       [[72, 76, 85, 93],
        [66, 34, 80, 12],
        [99, 17, 98, 23]]])
>>> for i in range(3):
...     print(a.max(i))
... 
[[72 76 85 93]
 [74 59 80 32]
 [99 94 98 32]]
[[77 94 81 32]
 [99 76 98 93]]
[[36 79 94]
 [93 80 99]]

可以见得：
ａ是一个2x3x4的三维矩阵。
当a.max(0)时，max则在维度大小为２的方向上进行操作，所以
a.max(0)就是：
[[72 76 85 93]
[74 59 80 32]
[99 94 98 32]]
一个　１x３x４的矩阵。
以此类推，a.max(1)就是在维度大小为３的方向上进行操作
a.max(i)就是：
[[77 94 81 32]
[99 76 98 93]]
一个　１ｘ２ｘ４的矩阵。

由此很容易发现。
.max(dim)中的dim，并不是ａ上的维度。而是指ａ的shape上的顺序(可以这么理解)，ａ的shape是２ｘ３ｘ４，也就是[2, 3, 4]。故可以这样一一对应以来。
而不用死记硬背那些０是对列操作还是对行操作了