Explicitly set number of output task dimensions