b2951813547426828d313a80c52de8a619e99731,gpytorch/lazy/constant_mul_lazy_tensor.py,ConstantMulLazyTensor,_t_matmul,#ConstantMulLazyTensor#Any#,85

Before Change

    def _t_matmul(self, rhs):
        res = self.base_lazy_tensor._t_matmul(rhs)
        res = res * self._constant_as(res)
        return res

    def _quad_form_derivative(self, left_vecs, right_vecs):
        res = list(self.base_lazy_tensor._quad_form_derivative(left_vecs, right_vecs))

After Change

    def _t_matmul(self, rhs):
        res = self.base_lazy_tensor._t_matmul(rhs)
        constant = self.constant.view(*self.constant.shape, 1, 1)
        res = res * constant
        return res

    def _quad_form_derivative(self, left_vecs, right_vecs):
        // Gradient with respect to the constant

Instances