从例子可以看出 对于一个 2*2*2*5 的张量组 最浅显的理解
如果 stack 从0维合并 可以将输入看作 [a, b] 进行合并如果 stack 从1维合并 可以将输入看作 [ [a, b] , [c, d] ] 进行合并如果 stack 从2维合并 可以将输入看作 [ [ [a, b], [c, d] ], [ [e, f], [g, h] ] ] 进行合并如果 stack 从3维合并 可以将输入看作[ [ [ [a1, a2, a3, a4, a5], [b1, b2, b3, b4, b5] ],
[ [c1, c2, c3, c4, c5], [d1, d2, d3, d4, d5] ] ],
[ [ [e1, e2, e3, e4, e5], [f1, f2, f3, f4, f5] ],
[ [g1, g2, g3, g4, g5], [h1, h2, h3, h4, h5] ] ] ] 进行合并。
也就说 随着axis的增加 stack方法合并的维度就在增加 从表层维度合并到更深层次的维度。这样就比较好理解stack方法了。
