f0eea66ce83fc59910c623aa4e50ddfdd1f3ae7b,gpytorch/lazy/matmul_lazy_variable.py,MatmulLazyVariable,evaluate,#MatmulLazyVariable#,96

Before Change

            return super(MatmulLazyVariable, self).diag()

    def evaluate(self):
        return torch.matmul(self.lhs.evaluate(), self.rhs.evaluate())

After Change

            return super(MatmulLazyVariable, self).diag()

    def evaluate(self):
        if not hasattr(self, "_evaluated_memo"):
            self._evaluated_memo = torch.matmul(
                self._evaluated_lhs,
                self._evaluated_rhs,
            )
        return self._evaluated_memo

Instances