2018-06-26 numpy的axis

数组转置

分步求导工具

How does NumPy's transpose() method permute the axes of an array?

Python · numpy · axis

如果我们换一种思路的话，以 axis=0 为例，由于我们现在整个数组里面一共有 24 个数，而 axis=0 只有两个元素，所以可以理解为在 axis=0 这个 axis 上，每隔 24 / 2 = 12 个数就跳一下。比如说上面这个图中就可以看出，两个橙色矩阵对应的数之间差的都是 12

类似的，由于一个橙色矩阵中只有 24 / 2 = 12 个数，所以我们可以理解为在 axis=1 这个 axis 上，每隔 12 / 3 = 4 个数就跳一下。表现在图中，就是同一个橙色矩阵的两个相邻的蓝色向量对应的数之间差的都是 4

再次类似的，由于一个蓝色向量中只有 12 / 3 = 4 个数，我们可以理解为在 axis=2 这个 axis 上，每隔 4 / 4 = 1 个数就跳一下。表现在图中……观众老爷们想必也知道是怎样的了 ( σ'ω')σ

所以我们现在可以定义一个新的东西，比如说叫做 strides 吧，它记录着每个 axis 上跳过的数。比如说上图对应的三维数组，它的 strides 就是 (12, 4, 1)

那么接下来激动人心的时刻到了：transpose 的本质，其实就是对 strides 中各个数的顺序进行调换。举个例子：