Function setup

benchmarks/emd.py:9–16

(n_samples)

def setup(n_samples):
    rng = np.random.RandomState(789465132)
    x = rng.randn(n_samples, 2)
    y = rng.randn(n_samples, 2)

    a = ot.utils.unif(n_samples)
    M = ot.dist(x, y)
    return a, M


if __name__ == "__main__":