all_solves = base_lazy_tensor._solve(torch.cat(eager_rhss, -1), base_lazy_tensor._preconditioner()[0])
solves = []
for eager_rhs in eager_rhss:
solve = all_solves[..., :eager_rhs.size(-1)]
all_solves = all_solves[..., eager_rhs.size(-1):]
solves.append(solve.detach()) // The detach is necessary here for some reason?super(CachedCGLazyTensor, self).__init__(
base_lazy_tensor, eager_rhss=eager_rhss, solves=solves,