ut3_probe_corewise_transpose#
- t3toolbox.backend.ut3_sampling.ut3_probe_corewise_transpose(ztildes, ww, data, sum_over_probes=False)#
def ut3_probe_corewise_transpose( ztildes: typ.Sequence[NDArray], # probe residuals, len=d, ith elm_shape=W+K+C+(Ni,) ww: typ.Sequence[NDArray], # probe vectors, len=d, ith elm_shape=W+(Ni,) data: UT3Data, sum_over_probes: bool = False, ) -> typ.Tuple[NDArray, NDArray]: # (tucker-core grad supercore, tt-core grad supercore)
Corewise transpose of
ut3_probe(): gradient w.r.t. the frame supercores.ztildesandwwpacked to N.