Index A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | Q | R | S | T | U | V | W | X A a (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) act_fn (ott.neural.networks.icnn.ICNN attribute) act_fn() (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) all_pairs() (ott.geometry.costs.Arccos method) (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SoftDTW method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.UnbalancedBures method) (ott.geometry.distrib_costs.UnivariateWasserstein method) all_pairs_pairwise() (ott.geometry.costs.Arccos method) (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SoftDTW method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.UnbalancedBures method) (ott.geometry.distrib_costs.UnivariateWasserstein method) AndersonAcceleration (class in ott.solvers.linear.acceleration) animate() (ott.tools.plot.Plot method) apply() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) apply_cost() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_fn (ott.neural.networks.potentials.PotentialTrainState attribute) apply_gradients() (ott.neural.networks.potentials.PotentialTrainState method) apply_kernel() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_lse_kernel() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_square_cost() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_transport_from_potentials() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_transport_from_scalings() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) Arccos (class in ott.geometry.costs) assignment (ott.tools.k_means.KMeansOutput attribute) at() (ott.geometry.epsilon_scheduler.Epsilon method) B b (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) barycenter() (ott.geometry.costs.Arccos method) (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SoftDTW method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.UnbalancedBures method) (ott.geometry.distrib_costs.UnivariateWasserstein method) (ott.geometry.pointcloud.PointCloud method) BaseFlow (class in ott.neural.methods.flows.dynamics) BasePotential (class in ott.neural.networks.potentials) batch_size (ott.geometry.pointcloud.PointCloud property) bias (ott.geometry.low_rank.LRCGeometry property) bias_init() (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) BrownianBridge (class in ott.neural.methods.flows.dynamics) Bures (class in ott.geometry.costs) C can_LRC (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) centroids (ott.tools.k_means.KMeansOutput attribute) cholesky (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) clone() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) component_weight_ob (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) component_weights (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) components() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) compute_error() (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) compute_kl_reg_cost() (ott.solvers.linear.sinkhorn.SinkhornState method) compute_max_cost() (ott.geometry.low_rank.LRCGeometry method) compute_mu_t() (ott.neural.methods.flows.dynamics.BaseFlow method) (ott.neural.methods.flows.dynamics.BrownianBridge method) (ott.neural.methods.flows.dynamics.ConstantNoiseFlow method) (ott.neural.methods.flows.dynamics.StraightFlow method) compute_reg_gw_cost() (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) compute_reg_ot_cost() (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) compute_sigma_t() (ott.neural.methods.flows.dynamics.BaseFlow method) (ott.neural.methods.flows.dynamics.BrownianBridge method) (ott.neural.methods.flows.dynamics.ConstantNoiseFlow method) (ott.neural.methods.flows.dynamics.StraightFlow method) compute_ut() (ott.neural.methods.flows.dynamics.BaseFlow method) (ott.neural.methods.flows.dynamics.BrownianBridge method) (ott.neural.methods.flows.dynamics.ConstantNoiseFlow method) (ott.neural.methods.flows.dynamics.StraightFlow method) compute_xt() (ott.neural.methods.flows.dynamics.BaseFlow method) (ott.neural.methods.flows.dynamics.BrownianBridge method) (ott.neural.methods.flows.dynamics.ConstantNoiseFlow method) (ott.neural.methods.flows.dynamics.StraightFlow method) condition (ott.neural.datasets.OTData attribute) condition_dims (ott.neural.networks.velocity_field.VelocityField attribute) conditional_log_prob() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) ConjugateResults (class in ott.neural.networks.layers.conjugate) ConstantNoiseFlow (class in ott.neural.methods.flows.dynamics) converged (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) (ott.tools.k_means.KMeansOutput attribute) copy() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) copy_epsilon() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) Cosine (class in ott.geometry.costs) cost (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) cost_1 (ott.geometry.low_rank.LRCGeometry property) cost_2 (ott.geometry.low_rank.LRCGeometry property) cost_at_geom() (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) cost_matrix (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) cost_rank (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) cost_unbalanced_correction() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) CostFn (class in ott.geometry.costs) costs (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) costs_bary (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) count() (ott.neural.networks.layers.conjugate.ConjugateResults method) (ott.problems.quadratic.quadratic_costs.GWLoss method) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState method) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn.SinkhornState method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState method) (ott.tools.k_means.KMeansOutput method) covariance (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) covariance() (ott.tools.gaussian_mixture.gaussian.Gaussian method) covariance_fixpoint_iter() (ott.geometry.costs.Bures method) create() (ott.neural.networks.potentials.PotentialTrainState class method) create_initializer() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) create_train_state() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) crossed_threshold (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) cyclical_time_encoder() (in module ott.neural.networks.layers.time_encoder) D default_progress_fn() (in module ott.utils) DefaultInitializer (class in ott.initializers.linear.initializers) dim_data (ott.neural.networks.icnn.ICNN attribute) dim_hidden (ott.neural.networks.icnn.ICNN attribute) (ott.neural.networks.layers.posdef.PositiveDense attribute) (ott.neural.networks.potentials.PotentialMLP attribute) distance() (ott.problems.linear.potentials.DualPotentials method) (ott.problems.linear.potentials.EntropicPotentials method) done() (ott.geometry.epsilon_scheduler.Epsilon method) done_at() (ott.geometry.epsilon_scheduler.Epsilon method) dtype (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) dual_cost (ott.solvers.linear.sinkhorn.SinkhornOutput property) DualPotentials (class in ott.problems.linear.potentials) dykstra_update_kernel() (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) dykstra_update_lse() (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) E ElasticL1 (class in ott.geometry.costs) ElasticL2 (class in ott.geometry.costs) ElasticSqKOverlap (class in ott.geometry.costs) ElasticSTVS (class in ott.geometry.costs) ent_reg_cost (ott.solvers.linear.sinkhorn.SinkhornOutput property) EntropicPotentials (class in ott.problems.linear.potentials) Epsilon (class in ott.geometry.epsilon_scheduler) epsilon (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.linear.potentials.EntropicPotentials property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) error (ott.tools.k_means.KMeansOutput attribute) error_threshold (ott.solvers.linear.acceleration.Momentum attribute) errors (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornState attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) Euclidean (class in ott.geometry.costs) extrapolation() (ott.solvers.linear.acceleration.AndersonAcceleration method) F f (ott.problems.linear.potentials.DualPotentials property) (ott.problems.linear.potentials.EntropicPotentials property) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput property) f1 (ott.problems.quadratic.quadratic_costs.GWLoss attribute) f2 (ott.problems.quadratic.quadratic_costs.GWLoss attribute) f_potential() (ott.tools.gaussian_mixture.gaussian.Gaussian method) FenchelConjugateLBFGS (class in ott.neural.networks.layers.conjugate) FenchelConjugateSolver (class in ott.neural.networks.layers.conjugate) first_order_conditions() (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) fit_model_em() (in module ott.tools.gaussian_mixture.fit_gmm) fitting_loss (ott.neural.methods.monge_gap.MongeGapEstimator property) FixedBarycenter (class in ott.solvers.linear.discrete_barycenter) FixedBarycenterProblem (class in ott.problems.linear.barycenter_problem) fixpoint_iter() (in module ott.math.fixed_point_loop) flattened_b (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) flattened_y (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) FreeBarycenterProblem (class in ott.problems.linear.barycenter_problem) FreeBarycenterState (class in ott.solvers.linear.continuous_barycenter) FreeWassersteinBarycenter (class in ott.solvers.linear.continuous_barycenter) from_graph() (ott.geometry.geodesic.Geodesic class method) (ott.geometry.graph.Graph class method) from_mean_and_cov() (ott.tools.gaussian_mixture.gaussian.Gaussian class method) from_mean_cov_component_weights() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture class method) from_pointcloud() (ott.geometry.low_rank.LRKGeometry class method) from_points_and_assignment_probs() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture class method) from_random() (ott.tools.gaussian_mixture.gaussian.Gaussian class method) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture class method) from_samples() (ott.tools.gaussian_mixture.gaussian.Gaussian class method) from_solver() (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer class method) (ott.initializers.linear.initializers_lr.KMeansInitializer class method) (ott.initializers.linear.initializers_lr.LRInitializer class method) (ott.initializers.linear.initializers_lr.RandomInitializer class method) (ott.initializers.linear.initializers_lr.Rank2Initializer class method) from_z() (ott.tools.gaussian_mixture.gaussian.Gaussian method) fu (ott.solvers.linear.sinkhorn.SinkhornState property) G g (ott.problems.linear.potentials.DualPotentials property) (ott.problems.linear.potentials.EntropicPotentials property) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) gamma (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) Gaussian (class in ott.tools.gaussian_mixture.gaussian) gaussian_map_samples (ott.neural.networks.icnn.ICNN attribute) GaussianInitializer (class in ott.initializers.linear.initializers) GaussianMixture (class in ott.tools.gaussian_mixture.gaussian_mixture) GaussianMixturePair (class in ott.tools.gaussian_mixture.gaussian_mixture_pair) GeneralizedKMeansInitializer (class in ott.initializers.linear.initializers_lr) GENOT (class in ott.neural.methods.flows.genot) Geodesic (class in ott.geometry.geodesic) geom (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) geom_xx (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) geom_xy (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) geom_yy (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) geometries (ott.geometry.grid.Grid property) Geometry (class in ott.geometry.geometry) get_bures_geometry() (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair method) get_component() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) get_cost_matrix() (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair method) get_fit_model_em_fn() (in module ott.tools.gaussian_mixture.fit_gmm_pair) get_log_component_posterior() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) get_normalized_sinkhorn_coupling() (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair method) get_sinkhorn() (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair method) get_step_fn() (ott.neural.methods.neuraldual.W2NeuralDual method) get_transport_functions() (ott.problems.linear.linear_problem.LinearProblem method) get_variable() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) gmm0 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) gmm1 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) grad (ott.neural.networks.layers.conjugate.ConjugateResults attribute) gradient() (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) Graph (class in ott.geometry.graph) Grid (class in ott.geometry.grid) GromovWasserstein (class in ott.solvers.quadratic.gromov_wasserstein) GromovWassersteinBarycenter (class in ott.solvers.quadratic.gw_barycenter) gtol (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) gv (ott.solvers.linear.sinkhorn.SinkhornState property) gw_convergence (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) gw_loss (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) GWBarycenterProblem (class in ott.problems.quadratic.gw_barycenter) GWBarycenterState (class in ott.solvers.quadratic.gw_barycenter) GWLoss (class in ott.problems.quadratic.quadratic_costs) GWOutput (class in ott.solvers.quadratic.gromov_wasserstein) H h() (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) h1 (ott.problems.quadratic.quadratic_costs.GWLoss attribute) h2 (ott.problems.quadratic.quadratic_costs.GWLoss attribute) h_legendre() (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) h_transform() (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) has_nans() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) has_rng() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) has_variable() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) hidden_dims (ott.neural.networks.velocity_field.VelocityField attribute) histogram (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) I ICNN (class in ott.neural.networks.icnn) ImplicitDiff (class in ott.solvers.linear.implicit_differentiation) increase_factor (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) index() (ott.neural.networks.layers.conjugate.ConjugateResults method) (ott.problems.quadratic.quadratic_costs.GWLoss method) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState method) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn.SinkhornState method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState method) (ott.tools.k_means.KMeansOutput method) init() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) init_dual_a() (ott.initializers.linear.initializers.DefaultInitializer method) (ott.initializers.linear.initializers.GaussianInitializer method) (ott.initializers.linear.initializers.SortingInitializer method) (ott.initializers.linear.initializers.SubsampleInitializer method) init_dual_b() (ott.initializers.linear.initializers.DefaultInitializer method) (ott.initializers.linear.initializers.GaussianInitializer method) (ott.initializers.linear.initializers.SortingInitializer method) (ott.initializers.linear.initializers.SubsampleInitializer method) init_fn() (ott.neural.networks.icnn.ICNN method) init_from_samples() (ott.neural.networks.layers.posdef.PosDefPotentials class method) init_g() (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer method) (ott.initializers.linear.initializers_lr.KMeansInitializer method) (ott.initializers.linear.initializers_lr.LRInitializer method) (ott.initializers.linear.initializers_lr.RandomInitializer method) (ott.initializers.linear.initializers_lr.Rank2Initializer method) init_maps() (ott.solvers.linear.acceleration.AndersonAcceleration method) init_q() (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer method) (ott.initializers.linear.initializers_lr.KMeansInitializer method) (ott.initializers.linear.initializers_lr.LRInitializer method) (ott.initializers.linear.initializers_lr.RandomInitializer method) (ott.initializers.linear.initializers_lr.Rank2Initializer method) init_r() (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer method) (ott.initializers.linear.initializers_lr.KMeansInitializer method) (ott.initializers.linear.initializers_lr.LRInitializer method) (ott.initializers.linear.initializers_lr.RandomInitializer method) (ott.initializers.linear.initializers_lr.Rank2Initializer method) init_state() (ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter method) (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter method) init_transport_mass() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) init_with_output() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) initialize() (in module ott.tools.gaussian_mixture.fit_gmm) inner_errors (ott.tools.k_means.KMeansOutput attribute) inner_iterations (ott.solvers.linear.acceleration.Momentum attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) inv_scale_cost (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) is_balanced (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) is_debiased (ott.problems.linear.potentials.EntropicPotentials property) is_equal_size (ott.problems.linear.linear_problem.LinearProblem property) is_fused (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) is_initializing() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) is_low_rank (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) (ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter property) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein property) (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter property) (ott.solvers.was_solver.WassersteinSolver property) is_mutable_collection() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) is_online (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) is_potential (ott.neural.networks.icnn.ICNN property) (ott.neural.networks.potentials.BasePotential property) (ott.neural.networks.potentials.PotentialMLP attribute) is_squared_euclidean (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) is_symmetric (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) is_uniform (ott.problems.linear.linear_problem.LinearProblem property) iteration (ott.tools.k_means.KMeansOutput attribute) K k_means() (in module ott.tools.k_means) kernel_diag_init() (ott.neural.networks.layers.posdef.PosDefPotentials method) kernel_init() (ott.neural.networks.layers.posdef.PositiveDense method) kernel_linear_init() (ott.neural.networks.layers.posdef.PosDefPotentials method) kernel_lr_init() (ott.neural.networks.layers.posdef.PosDefPotentials method) kernel_matrix (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) kernel_step() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) kl_reg_cost (ott.solvers.linear.sinkhorn.SinkhornOutput property) KMeansInitializer (class in ott.initializers.linear.initializers_lr) KMeansOutput (class in ott.tools.k_means) L lambertw (in module ott.math.utils) lazy_init() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) lehmann() (ott.solvers.linear.acceleration.Momentum method) lin (ott.neural.datasets.OTData attribute) linear_convergence (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) linear_loss (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) linear_state (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) LinearProblem (class in ott.problems.linear.linear_problem) linesearch_init (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) linesearch_type (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) loc (ott.tools.gaussian_mixture.gaussian.Gaussian property) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) lock_gmm1 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) log_component_weights() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) log_prob() (ott.tools.gaussian_mixture.gaussian.Gaussian method) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) logsumexp (in module ott.math.utils) LRCGeometry (class in ott.geometry.low_rank) LRGromovWasserstein (class in ott.solvers.quadratic.gromov_wasserstein_lr) LRGWOutput (class in ott.solvers.quadratic.gromov_wasserstein_lr) LRInitializer (class in ott.initializers.linear.initializers_lr) LRKGeometry (class in ott.geometry.low_rank) LRSinkhorn (class in ott.solvers.linear.sinkhorn_lr) LRSinkhornOutput (class in ott.solvers.linear.sinkhorn_lr) LRSinkhornState (class in ott.solvers.linear.sinkhorn_lr) lse_step() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) M make_kl_loss() (in module ott.problems.quadratic.quadratic_costs) make_rng() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) make_square_loss() (in module ott.problems.quadratic.quadratic_costs) marginal() (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) marginal_dependent_cost() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) marginal_from_potentials() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) marginal_from_scalings() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) mask() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) match_linear() (in module ott.solvers.utils) match_quadratic() (in module ott.solvers.utils) matrix (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) max_iter (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) max_linesearch_iter (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) max_measure_size (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) mean_cost_matrix (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) median_cost_matrix (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) memory (ott.solvers.linear.acceleration.AndersonAcceleration attribute) module ott ott.geometry ott.initializers ott.initializers.linear ott.initializers.quadratic ott.math ott.neural ott.neural.datasets ott.neural.methods ott.neural.methods.flows ott.neural.networks ott.neural.networks.layers ott.problems ott.problems.linear ott.problems.quadratic ott.solvers ott.solvers.linear ott.solvers.quadratic ott.tools ott.tools.gaussian_mixture ott.utils Momentum (class in ott.solvers.linear.acceleration) monge_gap() (in module ott.neural.methods.monge_gap) monge_gap_from_samples() (in module ott.neural.methods.monge_gap) MongeGapEstimator (class in ott.neural.methods.monge_gap) multivariate_cdf_quantile_maps() (in module ott.tools.soft_sort) N n_components (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) n_dimensions (ott.tools.gaussian_mixture.gaussian.Gaussian property) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) n_iters (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState property) name (ott.neural.networks.icnn.ICNN attribute) (ott.neural.networks.layers.posdef.PosDefPotentials attribute) (ott.neural.networks.layers.posdef.PositiveDense attribute) (ott.neural.networks.potentials.BasePotential attribute) (ott.neural.networks.potentials.PotentialMLP attribute) (ott.neural.networks.velocity_field.VelocityField attribute) ndim (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) ndim_fused (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) norm (in module ott.math.utils) (ott.geometry.costs.Arccos attribute) (ott.geometry.costs.Cosine attribute) (ott.geometry.costs.CostFn attribute) (ott.geometry.costs.ElasticL1 attribute) (ott.geometry.costs.ElasticL2 attribute) (ott.geometry.costs.ElasticSqKOverlap attribute) (ott.geometry.costs.ElasticSTVS attribute) (ott.geometry.costs.Euclidean attribute) (ott.geometry.costs.PNormP attribute) (ott.geometry.costs.SoftDTW attribute) (ott.geometry.costs.SqPNorm attribute) (ott.geometry.distrib_costs.UnivariateWasserstein attribute) norm() (ott.geometry.costs.Bures method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.UnbalancedBures method) norm_error (ott.solvers.linear.sinkhorn.Sinkhorn property) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein property) num_iter (ott.neural.networks.layers.conjugate.ConjugateResults attribute) num_measures (ott.problems.linear.barycenter_problem.FixedBarycenterProblem property) (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) num_potentials (ott.neural.networks.layers.posdef.PosDefPotentials attribute) num_quantiles (ott.solvers.linear.univariate.UnivariateSolver property) O old_fus (ott.solvers.linear.sinkhorn.SinkhornState attribute) old_mapped_fus (ott.solvers.linear.sinkhorn.SinkhornState attribute) old_transport_mass (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) one_iteration() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) opt_state (ott.neural.networks.potentials.PotentialTrainState attribute) ot_prob (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) OTData (class in ott.neural.datasets) OTDataset (class in ott.neural.datasets) OTFlowMatching (class in ott.neural.methods.flows.otfm) ott module ott.geometry module ott.initializers module ott.initializers.linear module ott.initializers.quadratic module ott.math module ott.neural module ott.neural.datasets module ott.neural.methods module ott.neural.methods.flows module ott.neural.networks module ott.neural.networks.layers module ott.problems module ott.problems.linear module ott.problems.quadratic module ott.solvers module ott.solvers.linear module ott.solvers.quadratic module ott.tools module ott.tools.gaussian_mixture module ott.utils module outer_iterations (ott.solvers.linear.sinkhorn.Sinkhorn property) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein property) output_dims (ott.neural.networks.velocity_field.VelocityField attribute) output_from_state() (ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter method) (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter method) P pairwise() (ott.geometry.costs.Arccos method) (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SoftDTW method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.UnbalancedBures method) (ott.geometry.distrib_costs.UnivariateWasserstein method) param() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) params (ott.neural.networks.potentials.PotentialTrainState attribute) parent (ott.neural.networks.icnn.ICNN attribute) (ott.neural.networks.layers.posdef.PosDefPotentials attribute) (ott.neural.networks.layers.posdef.PositiveDense attribute) (ott.neural.networks.potentials.BasePotential attribute) (ott.neural.networks.potentials.PotentialMLP attribute) (ott.neural.networks.velocity_field.VelocityField attribute) path (ott.neural.networks.icnn.ICNN property) (ott.neural.networks.layers.posdef.PosDefPotentials property) (ott.neural.networks.layers.posdef.PositiveDense property) (ott.neural.networks.potentials.BasePotential property) (ott.neural.networks.potentials.PotentialMLP property) (ott.neural.networks.velocity_field.VelocityField property) perturb() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) Plot (class in ott.tools.plot) plot_ot_map() (ott.problems.linear.potentials.DualPotentials method) (ott.problems.linear.potentials.EntropicPotentials method) plot_potential() (ott.problems.linear.potentials.DualPotentials method) (ott.problems.linear.potentials.EntropicPotentials method) PNormP (class in ott.geometry.costs) PointCloud (class in ott.geometry.pointcloud) pos_weights (ott.neural.networks.icnn.ICNN attribute) PosDefPotentials (class in ott.neural.networks.layers.posdef) PositiveDense (class in ott.neural.networks.layers.posdef) potential_from_scaling() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) potential_gradient_fn (ott.neural.networks.potentials.PotentialTrainState attribute) potential_gradient_fn() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) potential_value_fn (ott.neural.networks.potentials.PotentialTrainState attribute) potential_value_fn() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) PotentialMLP (class in ott.neural.networks.potentials) potentials (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornState attribute) PotentialTrainState (class in ott.neural.networks.potentials) precision (ott.neural.networks.layers.posdef.PosDefPotentials attribute) (ott.neural.networks.layers.posdef.PositiveDense attribute) precondition_fun (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) prepare_divergences() (ott.geometry.geodesic.Geodesic class method) (ott.geometry.geometry.Geometry class method) (ott.geometry.graph.Graph class method) (ott.geometry.grid.Grid class method) (ott.geometry.low_rank.LRCGeometry class method) (ott.geometry.low_rank.LRKGeometry class method) (ott.geometry.pointcloud.PointCloud class method) primal_cost (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) prox_legendre_reg() (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) prox_reg() (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) put_variable() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) Q q (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) quad (ott.neural.datasets.OTData attribute) quad_loss (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) QuadraticInitializer (class in ott.initializers.quadratic.initializers) QuadraticProblem (class in ott.problems.quadratic.quadratic_problem) quantile() (in module ott.tools.soft_sort) quantile_normalization() (in module ott.tools.soft_sort) quantiles (ott.solvers.linear.univariate.UnivariateSolver property) quantize() (in module ott.tools.soft_sort) R r (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) RandomInitializer (class in ott.initializers.linear.initializers_lr) rank (ott.geometry.low_rank.LRKGeometry property) (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer property) (ott.initializers.linear.initializers_lr.KMeansInitializer property) (ott.initializers.linear.initializers_lr.LRInitializer property) (ott.initializers.linear.initializers_lr.RandomInitializer property) (ott.initializers.linear.initializers_lr.Rank2Initializer property) (ott.neural.networks.layers.posdef.PosDefPotentials attribute) Rank2Initializer (class in ott.initializers.linear.initializers_lr) ranks (ott.neural.networks.icnn.ICNN attribute) ranks() (in module ott.tools.soft_sort) recenter() (ott.solvers.linear.sinkhorn.SinkhornState method) rectifier_fn (ott.neural.networks.icnn.ICNN attribute) (ott.neural.networks.layers.posdef.PosDefPotentials attribute) (ott.neural.networks.layers.posdef.PositiveDense attribute) refresh_every (ott.solvers.linear.acceleration.AndersonAcceleration attribute) reg() (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) reg_gw_cost (ott.solvers.quadratic.gromov_wasserstein.GWOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) reg_ot_cost (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) reg_ot_cost() (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) regularizer (ott.neural.methods.monge_gap.MongeGapEstimator property) replace() (ott.neural.networks.potentials.PotentialTrainState method) (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) rho (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) ridge_identity (ott.solvers.linear.acceleration.AndersonAcceleration attribute) S sample() (ott.tools.gaussian_mixture.gaussian.Gaussian method) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) sample_conditional() (in module ott.solvers.utils) sample_joint() (in module ott.solvers.utils) scale (ott.tools.gaussian_mixture.gaussian.Gaussian property) scale_params (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) scaling_from_potential() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) scalings (ott.solvers.linear.sinkhorn.SinkhornOutput property) scope (ott.neural.networks.icnn.ICNN attribute) (ott.neural.networks.layers.posdef.PosDefPotentials attribute) (ott.neural.networks.layers.posdef.PositiveDense attribute) (ott.neural.networks.potentials.BasePotential attribute) (ott.neural.networks.potentials.PotentialMLP attribute) (ott.neural.networks.velocity_field.VelocityField attribute) segment_point_cloud() (in module ott.geometry.segment) segment_sinkhorn() (in module ott.tools.segment_sinkhorn) segment_sinkhorn_divergence() (in module ott.tools.sinkhorn_divergence) segmented_y_b (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) segmented_y_fused (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) set() (ott.geometry.epsilon_scheduler.Epsilon method) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn.SinkhornState method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState method) set_cost() (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) set_scale_cost() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) setup() (ott.neural.methods.monge_gap.MongeGapEstimator method) (ott.neural.methods.neuraldual.W2NeuralDual method) (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) shape (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) Sinkhorn (class in ott.solvers.linear.sinkhorn) sinkhorn_divergence() (in module ott.tools.sinkhorn_divergence) SinkhornBarycenterOutput (class in ott.solvers.linear.discrete_barycenter) SinkhornOutput (class in ott.solvers.linear.sinkhorn) SinkhornState (class in ott.solvers.linear.sinkhorn) SoftDTW (class in ott.geometry.costs) softmin (in module ott.math.utils) solution_error() (ott.solvers.linear.sinkhorn.SinkhornState method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) solve() (in module ott.solvers.linear) (in module ott.solvers.quadratic) (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS method) (ott.neural.networks.layers.conjugate.FenchelConjugateSolver method) (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) solve_jax_cg() (in module ott.solvers.linear.implicit_differentiation) solve_lineax() (in module ott.solvers.linear.lineax_implicit) solver (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) solver_kwargs (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) sort() (in module ott.tools.soft_sort) sort_with() (in module ott.tools.soft_sort) SortingInitializer (class in ott.initializers.linear.initializers) sow() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) SqEuclidean (class in ott.geometry.costs) SqPNorm (class in ott.geometry.costs) sqrtm (in module ott.math.matrix_square_root) src_mask (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) SRC_PREFIX (ott.neural.datasets.OTDataset attribute) start (ott.solvers.linear.acceleration.Momentum attribute) step (ott.neural.networks.potentials.PotentialTrainState attribute) StraightFlow (class in ott.neural.methods.flows.dynamics) SubsampleInitializer (class in ott.initializers.linear.initializers) subset() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) symmetric (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) T target (ott.geometry.epsilon_scheduler.Epsilon property) tau (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) tgt_mask (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) TGT_PREFIX (ott.neural.datasets.OTDataset attribute) third_lower_bound() (in module ott.solvers.quadratic.lower_bound) threshold (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) time_dims (ott.neural.networks.velocity_field.VelocityField attribute) time_encoder() (ott.neural.networks.velocity_field.VelocityField method) to_dual_potentials() (ott.neural.methods.neuraldual.W2NeuralDual method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) to_low_rank() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) to_LRCGeometry() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) to_z() (ott.tools.gaussian_mixture.gaussian.Gaussian method) topk_mask() (in module ott.tools.soft_sort) tqdm_progress_fn() (in module ott.utils) train_map_estimator() (ott.neural.methods.monge_gap.MongeGapEstimator method) train_neuraldual_alternating() (ott.neural.methods.neuraldual.W2NeuralDual method) train_neuraldual_parallel() (ott.neural.methods.neuraldual.W2NeuralDual method) transport() (ott.neural.methods.flows.genot.GENOT method) (ott.neural.methods.flows.otfm.OTFlowMatching method) (ott.problems.linear.potentials.DualPotentials method) (ott.problems.linear.potentials.EntropicPotentials method) (ott.tools.gaussian_mixture.gaussian.Gaussian method) transport_cost_at_geom() (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) transport_from_potentials() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) transport_from_scalings() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) transport_mass (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) twist_operator() (ott.geometry.costs.Arccos method) (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.ElasticL1 method) (ott.geometry.costs.ElasticL2 method) (ott.geometry.costs.ElasticSqKOverlap method) (ott.geometry.costs.ElasticSTVS method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SoftDTW method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.UnbalancedBures method) (ott.geometry.distrib_costs.UnivariateWasserstein method) tx (ott.neural.networks.potentials.PotentialTrainState attribute) U unbalanced_dykstra_kernel() (in module ott.solvers.linear.lr_utils) unbalanced_dykstra_lse() (in module ott.solvers.linear.lr_utils) UnbalancedBures (class in ott.geometry.costs) unbind() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) uniform_sampler() (in module ott.solvers.utils) UnivariateSolver (class in ott.solvers.linear.univariate) UnivariateWasserstein (class in ott.geometry.distrib_costs) update() (ott.solvers.linear.acceleration.AndersonAcceleration method) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState method) (ott.tools.plot.Plot method) update_barycenter() (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem method) update_features() (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem method) update_history() (ott.solvers.linear.acceleration.AndersonAcceleration method) update_linearization() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) update_lr_geom() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) update_lr_linearization() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) update_potential() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) update_scaling() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) update_state() (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter method) use_bias (ott.neural.networks.layers.posdef.PosDefPotentials attribute) (ott.neural.networks.layers.posdef.PositiveDense attribute) use_linear (ott.neural.networks.layers.posdef.PosDefPotentials attribute) V val (ott.neural.networks.layers.conjugate.ConjugateResults attribute) value (ott.solvers.linear.acceleration.Momentum attribute) variable() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.VelocityField method) variables (ott.neural.networks.icnn.ICNN property) (ott.neural.networks.layers.posdef.PosDefPotentials property) (ott.neural.networks.layers.posdef.PositiveDense property) (ott.neural.networks.potentials.BasePotential property) (ott.neural.networks.potentials.PotentialMLP property) (ott.neural.networks.velocity_field.VelocityField property) vec_apply_cost() (ott.geometry.pointcloud.PointCloud method) VelocityField (class in ott.neural.networks.velocity_field) W w2_dist() (ott.tools.gaussian_mixture.gaussian.Gaussian method) W2NeuralDual (class in ott.neural.methods.neuraldual) warm_start (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein property) WassersteinSolver (class in ott.solvers.was_solver) weight() (ott.solvers.linear.acceleration.Momentum method) weights (ott.problems.linear.barycenter_problem.FixedBarycenterProblem property) (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) X x (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute)