Set number of dimensions of output task