38a40828ec688be626e51ebcb1a60a54479fd997,gpytorch/lazy/toeplitz_lazy_variable.py,ToeplitzLazyVariable,diag,#ToeplitzLazyVariable#,53

Before Change

        else:
            WTW_diag = self.c[0].expand_as(self.c)

        if self.added_diag is not None:
            if len(self.added_diag) > len(WTW_diag):
                raise RuntimeError("Additional diagonal component length does not \
                                    match the rest of this implicit tensor.")
            WTW_diag = WTW_diag + self.added_diag

        return WTW_diag

    def evaluate(self):
        

After Change

        if len(self.J_left) != len(self.J_right):
            raise RuntimeError("diag not supported for non-square interpolated Toeplitz matrices.")
        WTW_diag = Variable(torch.zeros(len(self.J_right)))
        for i in range(len(self.J_right)):
            WTW_diag[i] = self[i:i + 1, i:i + 1].evaluate()

        return WTW_diag

    def evaluate(self):
        

Instances