If `transpose `is true, then `U` should be transposed before getting `n_steps `from `U.shape[1]`
If
transposeis true, thenUshould be transposed before gettingn_stepsfromU.shape[1]