module

transport.matrix

module torchmatch.transport.matrix

Cost-matrix optimal-transport solvers.

Takes a (B, N, M) cost matrix and returns a transport plan (or scalar divergence). The dispatcher pattern mirrors :mod:`torchmatch.assignment`. See ``docs/superpowers/specs/2026-05-21-transport-ops-design.md``.

Members

module

opsDirect handles for torch.ops.transport.* matrix-face ops.

type

Backend

function

marginal_errorCompute the max marginal error of a log-domain transport plan.
solve

Source: torchmatch/transport/matrix/__init__.py:1