quimb.experimental.tn_marginals

Helper functions for computing marginals of classical partition functions / SAT problems.

Functions

compute_all_marginals_via_slicing(tn, output_inds[, ...])
compute_all_marginals_via_slicing_shared(tn, output_inds)
compute_all_marginals_via_torch_autodiff(tn, output_inds)

Module Contents

quimb.experimental.tn_marginals.compute_all_marginals_via_slicing(tn, output_inds, optimize='auto-hq', **contract_kwargs)

quimb.experimental.tn_marginals.compute_all_marginals_via_slicing_shared(tn, output_inds, optimize='auto-hq', **contract_kwargs)

quimb.experimental.tn_marginals.compute_all_marginals_via_torch_autodiff(tn, output_inds, optimize='auto-hq', equalize_norms=1.0, contraction_width_error_threshold=float('inf'))