Fix 2D Lorenz with batch_size > 1

2nd try.

I think I fixed the 2D Lorenz version with a batch-size > 1. But I am really no expert in the Lorenz model so please check VERY carefully!

Close an old MR !60 (closed) and issue #75 (closed)