Skip to content

Commit 1db1744

Browse files
Merge pull request #1721 from npbaskerville/master
Swithed to tensor all method to avoid errors when using pyro samples.
2 parents 661359c + 8be71ec commit 1db1744

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gpytorch/lazy/root_lazy_tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def _matmul(self, rhs):
5656
return self.root._matmul(self.root._t_matmul(rhs))
5757

5858
def _mul_constant(self, constant):
59-
if constant > 0:
59+
if (constant > 0).all():
6060
res = self.__class__(self.root._mul_constant(constant.sqrt()))
6161
else:
6262
res = super()._mul_constant(constant)

0 commit comments

Comments
 (0)