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.