ut3_probe_corewise_derivatives_transpose

t3toolbox.backend.ut3_sampling.ut3_probe_corewise_derivatives_transpose(ztildes, ww, pp, data, order, sum_over_probes=False, chunk_size=sampling_derivatives.DEFAULT_CHUNK_SIZE)

def ut3_probe_corewise_derivatives_transpose(
        ztildes: typ.Sequence[NDArray],  # probe residual jets, len=d, ith elm_shape=(order+1,)+W+C+(Ni,)
        ww:      typ.Sequence[NDArray],  # probe vectors,          len=d, ith elm_shape=W+(Ni,)
        pp:      typ.Sequence[NDArray],  # perturbation vectors P, len=d, ith elm_shape=W+(Ni,)
        data:    UT3Data,
        order:   int,
        sum_over_probes: bool = False,
        chunk_size: typ.Optional[int] = sampling_derivatives.DEFAULT_CHUNK_SIZE,  # W-chunk size; None -> dense. docs/chunking.md
) -> typ.Tuple[NDArray, NDArray]:     # (tucker-core grad supercore, tt-core grad supercore)

Corewise transpose of ut3_probe_derivatives(): gradient w.r.t. the frame supercores. ztildes/ww/pp packed to N.

Parameters:

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

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

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

• data (UT3Data)

• order (int)

• sum_over_probes (bool)

• chunk_size (t3toolbox.backend.common.typ.Optional[int])

Return type:

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