pytc.kmat._stream_l_panels¶

pytc.kmat._stream_l_panels(U, phi_r, phi_s, panel_size)[source]¶

Iterate (U_panel, phi_r_panel, phi_s_panel) along axis-1 of U.

Each yielded panel has axis-1 size exactly panel_size (last panel is zero-padded to keep a single JIT shape). U panels are moved to device with jax.device_put when U lives on host.