Numpy中的dimshuffle等效函数

Alb*_*ert 9 python numpy theano

我经常使用theano.tensor.dimshuffle.Numpy有相同的功能吗?

我想我可以通过几个numpy.swapaxesnumpy.newaxis(对于广播维度)做同样的事情,numpy.reshape但有一些更简单或更直接的方式,就像dimshuffle

eic*_*erg 9

该功能numpy.transpose允许阵列轴的任何排列.

变体array.T是一种特殊情况,对应于array.transpose()没有参数,默认为array.transpose(range(array.ndim)[::-1]).

numpy.swapaxesnumpy.transpose限制于两个轴的排列.

theano.tensor.dimshuffle本质上对应于numpy.transpose,但另外,它允许通过'x'在应该创建轴的任何地方添加来创建用于广播的长度为1的新轴.在numpy的,这可以通过使用的组合来实现transposereshape.

请注意,在numpy中,transpose尽可能小心地返回数据视图.在theano中也可能是这种情况,但可能取决于代码的优化方式.