bfa0cd26654cda62436b3ff55247996094b0a85f,gpytorch/lazy/lazy_variable.py,LazyVariable,add_diag,#LazyVariable#Any#,69

Before Change

            - diag (Scalar Variable)
        
        from .diag_lazy_variable import DiagLazyVariable
        if len(self.size()) != 2 or (self.size()[0] != self.size()[1]):
            raise RuntimeError("add_diag only defined for square matrices")
        return self + DiagLazyVariable(diag.expand(self.size()[0]))

After Change

        from .diag_lazy_variable import DiagLazyVariable
        if self.size(-1) != self.size(-2):
            raise RuntimeError("add_diag only defined for square matrices")
        if self.ndimension() == 3:
            return self + DiagLazyVariable(diag.unsqueeze(0).expand(self.size(0), self.size(1)))
        else:
            return self + DiagLazyVariable(diag.expand(self.size(0)))

    def add_jitter(self):
        
        Adds jitter (i.e., a small diagonal component) to the matrix this LazyVariable represents.
        This could potentially be implemented as a no-op, however this could lead to numerical instabilities,

Instances