We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 661359c + 8be71ec commit 1db1744Copy full SHA for 1db1744
gpytorch/lazy/root_lazy_tensor.py
@@ -56,7 +56,7 @@ def _matmul(self, rhs):
56
return self.root._matmul(self.root._t_matmul(rhs))
57
58
def _mul_constant(self, constant):
59
- if constant > 0:
+ if (constant > 0).all():
60
res = self.__class__(self.root._mul_constant(constant.sqrt()))
61
else:
62
res = super()._mul_constant(constant)
0 commit comments