ott.solvers.linear.semidiscrete.SemidiscreteOutput.to_dataloader

SemidiscreteOutput.to_dataloader(rng, batch_size, **kwargs)

Create a semidiscrete dataloader.

Parameters:

• rng (Array) – Random number seed used for sampling from the source distribution.

• batch_size (int) – Batch size used in the dataloader to sample from source.

• kwargs (Any) – Keyword arguments for SemidiscreteDataloader.

Return type:

SemidiscreteDataloader

Returns:

The semidiscrete dataloader.