ott.neural.datasets

Contents

ott.neural.datasets#

The ott.neural.datasets contains datasets and needed for solving (conditional) neural optimal transport problems.

Datasets#

datasets.OTData([lin, quad, condition])

Distribution data for (conditional) optimal transport problems.

datasets.OTDataset(src_data, tgt_data[, ...])

Dataset for optimal transport problems.
Contents