pytc.solver.isdf_xtc_ccsd.contract_terms_t2_tucker

pytc.solver.isdf_xtc_ccsd.contract_terms_t2_tucker(t2, p, grad_p, u1, u3, d, u, z, *, occupied_pair_batch_size=8, rank_panel_size=128)

Factor-direct VVVV–T2 terms with rank-M Tucker X.

K1/K2/K3/D use the established factor-direct kernels. The two X terms consume U and Z directly and never reconstruct either dense X or a virtual four-index tile.

Parameters:

• t2 (jax.Array)

• p (jax.Array)

• grad_p (jax.Array)

• u1 (jax.Array)

• u3 (jax.Array)

• d (jax.Array)

• u (jax.Array)

• z (jax.Array)

• occupied_pair_batch_size (int)

• rank_panel_size (int)

Return type:

Mapping[str, jax.Array]