b0eecef1c8acdfe6e65f1e96ff6e26d2bc43e7db,gpytorch/lazy/diag_lazy_tensor.py,DiagLazyTensor,_quad_form_derivative,#DiagLazyTensor#Any#Any#,35

Before Change

    def _quad_form_derivative(self, left_vecs, right_vecs):
        res = left_vecs * right_vecs
        if res.ndimension() > self._diag.ndimension():
            res = res.sum(-1)
        return (res,)

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

After Change

    def _quad_form_derivative(self, left_vecs, right_vecs):
        res = left_vecs * right_vecs
        batch_shape = _mul_broadcast_shape(self._batch_shape, res.shape[:-1])
        if batch_shape != res.shape[:-1]:
            res = res.expand(batch_shape + res.shape[-1:])
        return (res,)

Instances