Use preconditioning for NonLazyTensors
@@ -48,9 +48,6 @@ class NonLazyTensor(LazyTensor):
def _get_indices(self, left_indices, right_indices):
return self.tensor[left_indices, right_indices]
- def add_diag(self, diag):
- return NonLazyTensor(add_diag(self.tensor, diag))
-
def _preconditioner(self):
# For a NonLazyTensor, it is intended to not use preconditioning, even when called for.
return None, None
