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.models.ICNN attribute) act_fn() (ott.neural.models.MLP 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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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_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) BaseW2NeuralDual (class in ott.neural.solvers.neuraldual) batch_size (ott.geometry.pointcloud.PointCloud property) bias (ott.geometry.low_rank.LRCGeometry property) bias_init() (ott.neural.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) 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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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_reg_gw_cost() (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) compute_reg_ot_cost() (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) conditional_log_prob() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) ConjugateResults (class in ott.neural.solvers.conjugate) 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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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.solvers.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_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.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) crossed_threshold (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) D default_progress_fn() (in module ott.utils) DefaultInitializer (class in ott.initializers.linear.initializers) dim_data (ott.neural.models.ICNN attribute) dim_hidden (ott.neural.layers.PositiveDense attribute) (ott.neural.models.ICNN attribute) (ott.neural.models.MLP 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.solvers.conjugate) FenchelConjugateSolver (class in ott.neural.solvers.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.solvers.map_estimator.MapEstimator 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.models.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) 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.solvers.neuraldual.W2NeuralDual method) get_transport_functions() (ott.problems.linear.linear_problem.LinearProblem method) get_variable() (ott.neural.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) gmm0 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) gmm1 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) grad (ott.neural.solvers.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.solvers.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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) has_variable() (ott.neural.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) histogram (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) I ICNN (class in ott.neural.models) ImplicitDiff (class in ott.solvers.linear.implicit_differentiation) increase_factor (ott.neural.solvers.conjugate.FenchelConjugateLBFGS attribute) index() (ott.neural.solvers.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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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) (ott.neural.models.MetaInitializer 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) (ott.neural.models.MetaInitializer method) init_fn() (ott.neural.models.ICNN method) init_from_samples() (ott.neural.layers.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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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.models.ICNN property) (ott.neural.models.MLP attribute) (ott.neural.solvers.neuraldual.BaseW2NeuralDual property) 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.layers.PosDefPotentials method) kernel_init() (ott.neural.layers.PositiveDense method) kernel_linear_init() (ott.neural.layers.PosDefPotentials method) kernel_lr_init() (ott.neural.layers.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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) lehmann() (ott.solvers.linear.acceleration.Momentum method) 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.solvers.conjugate.FenchelConjugateLBFGS attribute) linesearch_type (ott.neural.solvers.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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) make_square_loss() (in module ott.problems.quadratic.quadratic_costs) MapEstimator (class in ott.neural.solvers.map_estimator) 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) 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.solvers.conjugate.FenchelConjugateLBFGS attribute) max_linesearch_iter (ott.neural.solvers.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) MetaInitializer (class in ott.neural.models) MLP (class in ott.neural.models) module ott ott.geometry ott.initializers ott.initializers.linear ott.initializers.quadratic ott.math ott.neural ott.neural.solvers 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.losses) monge_gap_from_samples() (in module ott.neural.losses) 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.layers.PosDefPotentials attribute) (ott.neural.layers.PositiveDense attribute) (ott.neural.models.ICNN attribute) (ott.neural.models.MLP attribute) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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.solvers.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.layers.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) ot_prob (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) ott module ott.geometry module ott.initializers module ott.initializers.linear module ott.initializers.quadratic module ott.math module ott.neural module ott.neural.solvers 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_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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) parent (ott.neural.layers.PosDefPotentials attribute) (ott.neural.layers.PositiveDense attribute) (ott.neural.models.ICNN attribute) (ott.neural.models.MLP attribute) (ott.neural.solvers.neuraldual.BaseW2NeuralDual attribute) path (ott.neural.layers.PosDefPotentials property) (ott.neural.layers.PositiveDense property) (ott.neural.models.ICNN property) (ott.neural.models.MLP property) (ott.neural.solvers.neuraldual.BaseW2NeuralDual property) perturb() (ott.neural.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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.models.ICNN attribute) PosDefPotentials (class in ott.neural.layers) PositiveDense (class in ott.neural.layers) 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.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) potential_value_fn() (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) potentials (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornState attribute) precision (ott.neural.layers.PosDefPotentials attribute) (ott.neural.layers.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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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_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.layers.PosDefPotentials attribute) Rank2Initializer (class in ott.initializers.linear.initializers_lr) ranks (ott.neural.models.ICNN attribute) ranks() (in module ott.tools.soft_sort) recenter() (ott.solvers.linear.sinkhorn.SinkhornState method) rectifier_fn (ott.neural.layers.PosDefPotentials attribute) (ott.neural.layers.PositiveDense attribute) (ott.neural.models.ICNN 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.solvers.map_estimator.MapEstimator property) replace() (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) 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.layers.PosDefPotentials attribute) (ott.neural.layers.PositiveDense attribute) (ott.neural.models.ICNN attribute) (ott.neural.models.MLP attribute) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.map_estimator.MapEstimator method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) (ott.neural.solvers.neuraldual.W2NeuralDual 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.solvers.conjugate.FenchelConjugateLBFGS method) (ott.neural.solvers.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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual 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) start (ott.solvers.linear.acceleration.Momentum attribute) 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) third_lower_bound() (in module ott.solvers.quadratic.lower_bound) threshold (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) to_dual_potentials() (ott.neural.solvers.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.solvers.map_estimator.MapEstimator method) train_neuraldual_alternating() (ott.neural.solvers.neuraldual.W2NeuralDual method) train_neuraldual_parallel() (ott.neural.solvers.neuraldual.W2NeuralDual method) transport() (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) 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.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) UnivariateSolver (class in ott.solvers.linear.univariate) UnivariateWasserstein (class in ott.geometry.distrib_costs) update() (ott.neural.models.MetaInitializer method) (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.layers.PosDefPotentials attribute) (ott.neural.layers.PositiveDense attribute) use_linear (ott.neural.layers.PosDefPotentials attribute) V val (ott.neural.solvers.conjugate.ConjugateResults attribute) value (ott.solvers.linear.acceleration.Momentum attribute) variable() (ott.neural.layers.PosDefPotentials method) (ott.neural.layers.PositiveDense method) (ott.neural.models.ICNN method) (ott.neural.models.MLP method) (ott.neural.solvers.neuraldual.BaseW2NeuralDual method) variables (ott.neural.layers.PosDefPotentials property) (ott.neural.layers.PositiveDense property) (ott.neural.models.ICNN property) (ott.neural.models.MLP property) (ott.neural.solvers.neuraldual.BaseW2NeuralDual property) vec_apply_cost() (ott.geometry.pointcloud.PointCloud method) W w2_dist() (ott.tools.gaussian_mixture.gaussian.Gaussian method) W2NeuralDual (class in ott.neural.solvers.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)