轴的概念

红线框代表的是以轴为0相加的情况,而黄线则是轴为1,蓝线自不必多说就是轴为2的情况。
  具体到 numpy 中的多维数组来说,轴即是元素坐标的索引。比如,第0轴即是第1个索引,延0轴计算就是去掉坐标中的第一个索引。过程就是
1.遍历其他索引的所有可能组合
2. 取出一个组合,保持值不变,遍历第一个索引所有可能值
3.根据索引可以获得了同一个轴上的所有元素
4.对他们进行计算得到最后的元素
5.所有组合的最后结果组到一起就是最后的 n-1 维数组
所以如果一个多维数组的 shape 是 (a1, a2, a3, a4), 那么延轴0计算最后的数组shape 是 (a2, a3, a4), 延轴1计算最后的数组shape是 (a1, a3, a4)。
---------------------
作者:Exziro
来源:CSDN
原文:https://blog.csdn.net/Exziro/article/details/78152621
版权声明:本文为博主原创文章,转载请附上博文链接!

原文地址:https://www.cnblogs.com/jerryspace/p/10020285.html