如何在沒有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步，而左邊的軸有右邊的尺寸乘積給出的步幅。

這對你意味着：只要你收集相鄰的軸，它就會做“正確”的事情。