如何在没有ambuigity的情况下使用numpy.ndarray合并特定的轴

Question

基本上我想重塑由numpy.ndarray代表的张量。

例如，我想做这样的事情（乳胶符号）

A_ {i，j，k，l，m，n，p} - > A_ {i，jk，lm，np}

要么

A_ {i，j，k，l，m，n，p} - > A_ {ij，k，l，m，np}

其中A是ndarray。 i，j，k，...表示原始轴。

所以新的轴2成为轴2和3等的“扁平”版本。如果我只是使用numpy.reshape，我不认为它知道我想要合并的轴，所以它看起来很模糊且容易出错。

有没有任何巧妙的方法来做这个而不是手动创建另一个ndarray？

Answer 1

使用reshape永远不会模棱两可。 它不会改变数据的内存布局。

始终使用由形状确定的步幅完成索引。 最右边的轴有1步，而左边的轴有右边的尺寸乘积给出的步幅。

这对你意味着：只要你收集相邻的轴，它就会做“正确”的事情。