Numpy中的dimshuffle等价函数

Question

我经常用^{}。Numpy有等价的函数吗？

我想我也可以通过几个numpy.swapaxes和numpy.newaxis（广播维度），numpy.reshape来做同样的事情，但是有没有更简单或更直接的方法，就像dimshuffle？

Answer 1

函数numpy.transpose允许数组轴的任何排列。

变种array.T是这种情况的一个特例，对应于不带参数的array.transpose()，默认为array.transpose(range(array.ndim)[::-1])。

numpy.swapaxes仅限于两个轴的排列。

theano.tensor.dimshuffle本质上对应于numpy.transpose，但另外，它还允许创建长度为1的新轴用于广播，方法是在应创建轴的任何位置添加'x'。在numpy中，这可以通过结合使用transpose和reshape来实现。

注意，在numpy中，要注意尽可能让transpose返回数据视图。在不可能的情况下，可能也是这样，但可能取决于如何优化代码。