t3toolbox.backend.probing#

Functions#

probe_t3(, use_jax, t3toolbox.backend.common.NDArray])

Probe a Tucker tensor train.

compute_xis(...)

Compute upward edge variables associated with edges between Tucker cores and adjacent TT-cores.

compute_mus(...)

Compute leftward edge variables associated with edges between adjacent TT-cores.

compute_nus(...)

Compute rightward edge variables associated with edges between adjacent TT-cores.

compute_etas(...)

Compute downward edge variables associated with edges between Tucker cores and adjacent TT-cores.

assemble_zs(...)

Assemble probes from downward edge variables.

compute_dxis(...)

Compute var-upward edge variables dxi.

compute_sigmas(...)

Compute var-leftward edge variables sigma.

compute_taus(...)

Compute var-rightward edge variables tau.

compute_detas(...)

Compute var-downward edge variables deta.

assemble_tangent_zs(...)

Assemble tangent vector probes from edge variables.

probe_tangent(, use_jax, t3toolbox.backend.common.NDArray])

Probe a tangent vector.

compute_deta_tildes(...)

Adjoint-var-upward edge variables deta_tilde.

compute_tau_tildes(...)

Adjoint-var-rightward edge variables tau_tilde.

compute_sigma_tildes(...)

Adjoint-var-leftward edge variables sigma_tilde.

compute_dxi_tildes(...)

Adjoint-var-downward edge variables dxi_tilde.

assemble_tucker_variations(...)

Assemble Tucker backend variations, delta_U_tilde.

assemble_tt_variations(...)

Assemble TT backend variations, delta_G_tilde.

probe_tangent_transpose(, sum_over_probes, use_jax, ...)

Apply the transpose of the map from a T3Tangent to its probes. Apply to ztildes.

probe_dense(...)

Probe a dense tensor.