198493be78000034e4ce36d846bda0a82fd0779f,gpytorch/lazy/diag_lazy_tensor.py,DiagLazyTensor,_mul_matrix,#DiagLazyTensor#Any#,65

Before Change

    def _mul_matrix(self, other):
        if isinstance(other, DiagLazyTensor):
            return self.__class__(self._diag * other._diag)
        else:
            return self.__class__(self._diag * other.diag())

    def _prod_batch(self, dim):

After Change

        return self.__class__(self._diag * constant.unsqueeze(-1))

    def _mul_matrix(self, other):
        return DiagLazyTensor(self.diag() * other.diag())

    def _prod_batch(self, dim):
        return self.__class__(self._diag.prod(dim))

Instances