module

transport

module torchmatch.transport

Continuous optimal-transport (OT) solvers.

Two sub-packages: - :mod:`torchmatch.transport.matrix`: cost-matrix in, plan / divergence out. The primary dispatcher; mirrors :mod:`torchmatch.assignment`. - :mod:`torchmatch.transport.samples`: point-clouds in, scalar loss out. Backed by Triton kernels (CUDA-only). See ``docs/superpowers/specs/2026-05-21-transport-ops-design.md``.

Members

module

matrixCost-matrix optimal-transport solvers.
samplesSamples-face optimal transport.

function

load_cpuRegister ``torch.ops.transport.*`` (CPU backend).
load_cuda

Source: torchmatch/transport/__init__.py:1