d4806006e851ba3acf9d11d17eac9e90d41f487e,gpytorch/lazy/triangular_lazy_tensor.py,TriangularLazyTensor,add_diag,#TriangularLazyTensor#Any#,57

Before Change

    def add_diag(self, added_diag):
        shape = _mul_broadcast_shape(self._diag.shape, added_diag.shape)
        return self.__class__(
            AddedDiagLazyTensor(self._tensor.expand(shape), added_diag.expand(shape)), upper=self._upper
        )

    def diag(self):
        return self._tensor.diag()

After Change

        from .added_diag_lazy_tensor import AddedDiagLazyTensor

        shape = _mul_broadcast_shape(self._diag.shape, added_diag.shape)
        added_diag_lt = AddedDiagLazyTensor(self._tensor.expand(shape), added_diag.expand(shape))
        return TriangularLazyTensor(added_diag_lt, upper=self.upper)

    def diag(self) -> Tensor:
        return self._tensor.diag()

Instances