waxmorph.torch.losses.make_samples_loss

waxmorph.torch.losses.make_samples_loss#

waxmorph.torch.losses.make_samples_loss(params=None, **kwargs)[source]#

Return a lazily imported GeomLoss loss for unordered clouds.

params and kwargs merge with kwargs taking precedence. Unspecified options use GeomLoss defaults, explicit None is retained, and unknown keys raise TypeError. Supported families are Sinkhorn, Gaussian, Laplacian, energy, and Hausdorff. Hausdorff uses energy_kernel by default, including GeomLoss’s legacy import fallback. JAX’s factory provides OTT Sinkhorn divergence.

Parameters:

  • params (dict[str, Any] | None)

  • kwargs (Any)

Return type:

SamplesLoss