- r_barrel = (r_barrel + t_barrel + t0) % R
-
- # print(f"({N}, {H}, {R}, {t1-t0}) {G.size()=}")
-
- G = G[n_barrel, h_barrel, r_barrel, t_barrel]
-
- # print(G.sum())
+ r_barrel = (r_barrel + (t_barrel + t0) // L) % R
+ G = G.gather(dim=2, index=r_barrel.expand_as(G))