torchmatch
module

transport.samples

module torchmatch.transport.samples

Samples-face optimal transport.

Point clouds in, scalar OT cost / divergence out. Requires CUDA + Triton. See ``docs/superpowers/specs/2026-05-21-transport-ops-design.md``.

Members

module

  • kernelsTriton kernels for streaming optimal transport.

function

  • lossScalar OT cost / divergence over point clouds (CUDA only).

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