74db1a184a17bee6e658851d924c45740888f84a,gpytorch/lazy/constant_mul_lazy_variable.py,ConstantMulLazyVariable,_quad_form_derivative,#ConstantMulLazyVariable#Any#Any#,30

Before Change

            if torch.is_tensor(item) and res[i].sum():
                res[i] = res[i] * self.constant.expand_as(res[i])
        // Gradient with respect to the constant
        res.append(left_vecs.new(1).fill_((left_vecs * self.lazy_var._matmul(right_vecs)).sum()))
        return res

    def _size(self):

After Change

        if self.constant.numel() == 1:
            constant_deriv = (left_vecs * self.lazy_var._matmul(right_vecs)).sum()
        else:
            constant_deriv = (left_vecs * self.lazy_var._matmul(right_vecs)).sum(-2, keepdim=True).sum(-1, keepdim=True)

        res.append(constant_deriv)
        return res

Instances