pytc.solver.isdf_xtc_ccsd._contract_full_thc_t2_impl¶
- pytc.solver.isdf_xtc_ccsd._contract_full_thc_t2_impl(t2, a, b, z, c, e, *, occupied_pair_batch_size, rank_panel_size)[source]¶
JIT implementation of the blocked full-THC VVVV–T2 contraction.
- Parameters:
t2 (jax.Array)
a (jax.Array)
b (jax.Array)
z (jax.Array)
c (jax.Array)
e (jax.Array)
occupied_pair_batch_size (int)
rank_panel_size (int)
- Return type:
jax.Array