b53f2d7d650106a2ff175c2146b4ab897a3389d7,gpytorch/lazy/lazy_tensor.py,LazyTensor,diagonalization,#LazyTensor#Any#,1333

Before Change

            // from ..utils.lanczos import lanczos_tridiag, lanczos_tridiag_to_diag
            from ..lazy import lazify

            qmat, tmat = gpytorch.utils.lanczos.lanczos_tridiag(
                self.matmul,
                max_iter=settings.max_root_decomposition_size.value(),
                dtype=self.dtype,
                device=self.device,
                matrix_shape=self.shape[-2:],
                batch_shape=self.batch_shape,
            )
            evals, q_t = gpytorch.utils.lanczos.lanczos_tridiag_to_diag(tmat)
            evecs = lazify(qmat @ q_t)

        elif method == "symeig":

After Change

            from ..lazy import lazify

            func = Diagonalization()
            evals, evecs = func.apply(
                self.representation_tree(),
                self.device,
                self.dtype,

Instances