美文网首页
Numpy的axis图示理解——stack()中的axis

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

作者: 御风_曜 | 来源:发表于2020-11-06 15:03 被阅读0次

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。

相关文章

网友评论

      本文标题:Numpy的axis图示理解——stack()中的axis

      本文链接:https://www.haomeiwen.com/subject/iputbktx.html