Skip to content

Commit c58dc3a

Browse files
authored
Don't evaluate kernel in LazyEvaluatedKernelTensor._unsqueeze_batch (#1813)
Related to meta-pytorch/botorch#971.
1 parent ade5db8 commit c58dc3a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gpytorch/lazy/lazy_evaluated_kernel_tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ def add_jitter(self, jitter_val=1e-3):
237237
return self.evaluate_kernel().add_jitter(jitter_val)
238238

239239
def _unsqueeze_batch(self, dim):
240-
return self.evaluate_kernel()._unsqueeze_batch(dim)
240+
return self[(slice(None),) * dim + (None,)]
241241

242242
@cached(name="kernel_diag")
243243
def diag(self):

0 commit comments

Comments
 (0)