assemble_tucker_variation_jets_trs

t3toolbox.backend.sampling_derivatives.assemble_tucker_variation_jets_trs(ztildes, dxi_tildes, ww, pp, etas, n_probe, sum_over_probes)

def assemble_tucker_variation_jets_trs(
        ztildes:        typ.Sequence[NDArray],  # residual jets, len=d, elm_shape=(order+1,)+W+K+C+(Ni,)
        dxi_tildes:     typ.Sequence[NDArray],  # adjoint-var-down jets, len=d, elm_shape=(order+1,)+W+K+C+(nOi,)
        ww:             typ.Sequence[NDArray],  # probe vectors X,        len=d, elm_shape=W+(Ni,)
        pp:             typ.Sequence[NDArray],  # perturbation vectors P, len=d, elm_shape=W+(Ni,)
        etas:           typ.Sequence[NDArray],  # frame down jets, len=d, elm_shape=(order+1,)+W+C+(nOi,)
        n_probe:        int,                    # number of sample-stack (W) axes
        sum_over_probes: bool,
) -> typ.Tuple[NDArray, ...]:                   # dU_tildes. len=d, elm_shape=[W+]C+(nOi,Ni)

Assemble Tucker-core variation gradients (the 1-edge, plain-order-sum case): dU_tilde = sum_t eta^(t) (x) r^(t) + sum_u dxi_tilde^(u) (x) w_jet^(u).

Parameters:

• ztildes (t3toolbox.backend.common.typ.Sequence[NDArray])

• dxi_tildes (t3toolbox.backend.common.typ.Sequence[NDArray])

• ww (t3toolbox.backend.common.typ.Sequence[NDArray])

• pp (t3toolbox.backend.common.typ.Sequence[NDArray])

• etas (t3toolbox.backend.common.typ.Sequence[NDArray])

• n_probe (int)

• sum_over_probes (bool)

Return type:

t3toolbox.backend.common.typ.Tuple[NDArray, Ellipsis]