update dimshuffle