np.arange(1, n_times + 1, dtype=dtype)[None]

to avoid the +1 realloc