+ # insert_flash_back(
+ # self.rec_V,
+ # V,
+ # self.rec_K,
+ # K,
+ # t0,
+ # t1,
+ # CL,
+ # proba=self.proba_flashback / CL,
+ # )
+
+ n = torch.arange(N, device=X.device)[:, None, None, None]
+ t = torch.arange(t0, t1, device=X.device)[None, None, :, None]
+ dv = torch.arange(DV, device=X.device)[None, None, None, :]
+ dk = torch.arange(DK, device=X.device)[None, None, None, :]