pytc.kmat.contract_K1_minus_K2_isdf_streaming¶

pytc.kmat.contract_K1_minus_K2_isdf_streaming(phi_p, phi_q, phi_r, phi_s, grad_phi_p, grad_phi_q, U1, rank_block_size=128, panel_size=None)[source]¶

Streaming-capable wrapper around contract_K1_minus_K2_isdf_jit().

U1 can be a device jax.Array or a host numpy ndarray.