b2951813547426828d313a80c52de8a619e99731,gpytorch/lazy/constant_mul_lazy_tensor.py,ConstantMulLazyTensor,_quad_form_derivative,#ConstantMulLazyTensor#Any#Any#,90

Before Change

            constant_deriv = left_vecs * self.base_lazy_tensor._matmul(right_vecs)
            constant_deriv = constant_deriv.sum(-2, keepdim=True).sum(-1, keepdim=True)

        if constant_deriv.dim():
            constant_deriv = constant_deriv.view(*self.constant.size())
        res.append(constant_deriv)
        return res

    def _constant_as(self, other):

After Change

        constant_deriv = constant_deriv.sum(-2).sum(-1)

        // Get derivaties of everything else
        constant = self.constant.view(*self.constant.shape, 1, 1)
        left_vecs = left_vecs * constant
        res = self.base_lazy_tensor._quad_form_derivative(left_vecs, right_vecs)

        return res + (constant_deriv,)

Instances