bfc06c52e33e5d78bc4e9b332a5a7ac16fdcca46,gpytorch/lazy/mul_lazy_variable.py,MulLazyVariable,_lanczos_quadrature_form,#MulLazyVariable#,191

Before Change

        return closure

    def _lanczos_quadrature_form(self, *args):
        if not hasattr(self, "_lanczos_quadrature"):
            n = self.size()[0]
            z = args[0].new(n, 1).normal_()
            z = z / torch.norm(z, 2, 0)

            def tensor_matmul_closure(rhs):
                return self._matmul_closure_factory(*args)(rhs)

            Q, T = StochasticLQ(cls=type(z), max_iter=self.max_iter).lanczos_batch(tensor_matmul_closure, z)
            Q = Q[0]
            T = T[0]
            self._lanczos_quadrature = Q, T
        return self._lanczos_quadrature

    def _derivative_quadratic_form_factory(self, *args):
        args_index = []
        args_index.append(0)

After Change

        Q, T = StochasticLQ(cls=type(z), max_iter=self.max_iter).lanczos_batch(tensor_matmul_closure, z)
        Q = Q[0]
        T = T[0]
        return Q, T

    def _derivative_quadratic_form_factory(self, *args):
        args_index = []
        args_index.append(0)

Instances