4c6458b841bb49610b5cf6b9fbab94df5cad85ee,gpytorch/lazy/toeplitz_lazy_tensor.py,ToeplitzLazyTensor,_quad_form_derivative,#ToeplitzLazyTensor#Any#Any#,20

Before Change

        res = (sym_toeplitz_derivative_quadratic_form(left_vecs, right_vecs),)
        if self.column.ndimension() == 1 and res[0].ndimension() == 2:
            res = (res[0].sum(0),)
        return res

    def _size(self):
        if self.column.ndimension() == 2:

After Change

        res = sym_toeplitz_derivative_quadratic_form(left_vecs, right_vecs)

        // Collapse any expanded broadcast dimensions
        if res.dim() > self.column.dim():
            res = res.view(-1, *self.column.shape).sum(0)

        return res,

    def _size(self):
        return torch.Size((*self.column.shape, self.column.size(-1)))

Instances