link dimshuffle to swapaxes