correct writing of dimshuffle