utv_probe_transpose_derivatives_from_sweep#
- t3toolbox.backend.utv_sampling.utv_probe_transpose_derivatives_from_sweep(ztildes, frame_sweep, order, sum_over_probes=True, chunk_size=100)#
def utv_probe_transpose_derivatives_from_sweep( ztildes, # probe residual jets, len=d, ith elm_shape=(order+1,)+W+K+C+(Ni,) frame_sweep, # = utv_precompute_probe_frame_sweep_jets(...) order, # highest derivative order sum_over_probes=True, chunk_size=100, # W-chunk size for the gradient assembly; None -> dense. docs/chunking.md ): # -> bare variation supercore pair (dU_tilde, dG_tilde); stack K_new + C
Transpose
𝒥ᵀof the probe-derivatives reusing the jet sweep (the bare adjoint): pack the residual jets toN, sharesampling_derivatives.tv_probe_transpose_derivatives_from_sweep().