Avoid evaluating kernel in expand_batch (#2185)

cornellius-gp · Nov 29, 2022 · 93623d9 · 93623d9
1 parent 27b07e9
commit 93623d9
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/gpytorch/lazy/lazy_evaluated_kernel_tensor.py b/gpytorch/lazy/lazy_evaluated_kernel_tensor.py
@@ -111,9 +111,6 @@ def _diagonal(self) -> torch.Tensor:
             res = res.to_dense()
         return res.view(self.shape[:-1]).contiguous()
 
-    def _expand_batch(self, batch_shape):
-        return self.evaluate_kernel()._expand_batch(batch_shape)
-
     def _getitem(self, row_index, col_index, *batch_indices):
         x1 = self.x1
         x2 = self.x2