Numpy的axis图示理解——stack()中的axis

numpy.stack()比较难理解。其文档中的一个例子，如何理解axis=2的情况呢？

image.png

这个例子是用10 * 3 * 4张量。为了简化，我们先研究2 * 3 * 4。

2 * 3 * 4张量可以表示成如下。np.stack(axis=2)相当于在axis=2即第三维度的元素之间堆叠，图中用线连接起来就表示堆叠

image.png

因为图中的线连起来只有两个元素，所以堆叠的结果：

image.png

所以结果是一个3 * 4 * 2的张量。

同理可知10 * 3 * 4在stack(axis=2)时结果是3 * 4 * 10。