Index

A

A (ott.math.decomposition.DenseCholeskySolver property) (ott.math.decomposition.SparseCholeskySolver property) a (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) (ott.solvers.linear.continuous_barycenter.BarycenterState attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) act_fn (ott.solvers.nn.icnn.ICNN attribute) all_pairs() (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) (ott.geometry.costs.UnbalancedBures method) all_pairs_pairwise() (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) (ott.geometry.costs.UnbalancedBures method) AndersonAcceleration (class in ott.solvers.linear.acceleration) apply() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method)

apply_cost() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_kernel() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_lse_kernel() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_square_cost() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_transport_from_potentials() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_transport_from_scalings() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) 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) barycenter() (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) (ott.geometry.costs.UnbalancedBures method) (ott.geometry.pointcloud.PointCloud method)

BarycenterProblem (class in ott.problems.linear.barycenter_problem) BarycenterState (class in ott.solvers.linear.continuous_barycenter) batch_size (ott.geometry.pointcloud.PointCloud property) bias (ott.geometry.low_rank.LRCGeometry property) bias_init() (ott.solvers.nn.layers.PositiveDense method) bind() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) Bures (class in ott.geometry.costs)

C

can_LRC (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) centroids (ott.tools.k_means.KMeansOutput attribute) cholesky (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) clear_factor_cache() (ott.math.decomposition.SparseCholeskySolver class method) clone() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense 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_max_cost() (ott.geometry.low_rank.LRCGeometry method) compute_reg_ot_cost() (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) conditional_log_prob() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) converged (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.tools.k_means.KMeansOutput attribute) copy_epsilon() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry 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)

cost_matrix (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) cost_rank (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry 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.BarycenterState attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) count() (ott.solvers.linear.continuous_barycenter.BarycenterState method) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput 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.math.decomposition.DenseCholeskySolver class method) (ott.math.decomposition.SparseCholeskySolver class method) create_initializer() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein method) create_train_state() (ott.solvers.nn.icnn.ICNN method)

D

DefaultInitializer (class in ott.initializers.linear.initializers) DenseCholeskySolver (class in ott.math.decomposition) dim_data (ott.solvers.nn.icnn.ICNN attribute) dim_hidden (ott.solvers.nn.icnn.ICNN attribute) (ott.solvers.nn.layers.PositiveDense attribute) discrete_barycenter() (in module ott.solvers.linear.discrete_barycenter) 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.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry 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() (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method)

E

EntropicPotentials (class in ott.problems.linear.potentials) Epsilon (class in ott.geometry.epsilon_scheduler) epsilon (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.linear.potentials.EntropicPotentials property) (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.BarycenterState attribute) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput 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 attribute) f_potential() (ott.tools.gaussian_mixture.gaussian.Gaussian method) first_order_conditions() (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) fit_model_em() (in module ott.tools.gaussian_mixture.fit_gmm) fixpoint_iter() (in module ott.math.fixed_point_loop) flattened_b (ott.problems.linear.barycenter_problem.BarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) flattened_y (ott.problems.linear.barycenter_problem.BarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property)

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_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.RandomInitializer class method) (ott.initializers.linear.initializers_lr.Rank2Initializer class method) from_z() (ott.tools.gaussian_mixture.gaussian.Gaussian method)

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 attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) Gaussian (class in ott.tools.gaussian_mixture.gaussian) gaussian_map (ott.solvers.nn.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) geom (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) 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.solvers.nn.neuraldual.NeuralDualSolver method) get_transport_functions() (ott.problems.linear.linear_problem.LinearProblem method) get_variable() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) gmm0 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) gmm1 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) gradient() (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) Graph (class in ott.geometry.graph) graph (ott.geometry.graph.Graph property) Grid (class in ott.geometry.grid) GromovWasserstein (class in ott.solvers.quadratic.gromov_wasserstein) GromovWassersteinBarycenter (class in ott.solvers.quadratic.gw_barycenter) 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) GWOutput (class in ott.solvers.quadratic.gromov_wasserstein)

H

h() (ott.geometry.costs.PNormP method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) h_legendre() (ott.geometry.costs.PNormP method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method)

has_nans() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) has_rng() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) has_variable() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) histogram (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute)

I

ICNN (class in ott.solvers.nn.icnn) ImplicitDiff (class in ott.solvers.linear.implicit_differentiation) index() (ott.solvers.linear.continuous_barycenter.BarycenterState method) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState method) (ott.tools.k_means.KMeansOutput method) init() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) init_dual_a() (ott.initializers.linear.initializers.DefaultInitializer method) (ott.initializers.linear.initializers.GaussianInitializer method) (ott.initializers.linear.initializers.SinkhornInitializer method) (ott.initializers.nn.initializers.MetaInitializer method) init_dual_b() (ott.initializers.linear.initializers.DefaultInitializer method) (ott.initializers.linear.initializers.GaussianInitializer method) (ott.initializers.linear.initializers.SinkhornInitializer method) (ott.initializers.nn.initializers.MetaInitializer method) init_fn() (ott.solvers.nn.icnn.ICNN method) init_g() (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer method) (ott.initializers.linear.initializers_lr.KMeansInitializer 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.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.RandomInitializer method) (ott.initializers.linear.initializers_lr.Rank2Initializer method) init_state() (ott.solvers.linear.continuous_barycenter.WassersteinBarycenter method) (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein method) (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter method) init_std (ott.solvers.nn.icnn.ICNN attribute) init_transport_mass() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) init_with_output() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense 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) inv_rectifier_fn() (ott.solvers.nn.layers.PositiveDense method) inv_scale_cost (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry 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_entropic (ott.solvers.linear.sinkhorn_lr.LRSinkhorn property) is_fused (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) is_initializing() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) is_low_rank (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) (ott.solvers.linear.continuous_barycenter.WassersteinBarycenter 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.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) is_online (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) is_sparse (ott.geometry.graph.Graph property) is_squared_euclidean (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) is_symmetric (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) iteration (ott.tools.k_means.KMeansOutput attribute)

K

k_means() (in module ott.tools.k_means) kernel_init() (ott.solvers.nn.layers.PositiveDense method) kernel_matrix (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property)

kernel_step() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) KMeansInitializer (class in ott.initializers.linear.initializers_lr) KMeansOutput (class in ott.tools.k_means)

L

L (ott.math.decomposition.DenseCholeskySolver property) (ott.math.decomposition.SparseCholeskySolver property) laplacian (ott.geometry.graph.Graph property) lehmann() (ott.solvers.linear.acceleration.Momentum method) linear (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) linear_convergence (ott.solvers.linear.continuous_barycenter.BarycenterState attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) linear_loss (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) linear_output (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) linear_state (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute)

LinearProblem (class in ott.problems.linear.linear_problem) 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) LRCGeometry (class in ott.geometry.low_rank) LRQuadraticInitializer (class in ott.initializers.quadratic.initializers) LRSinkhorn (class in ott.solvers.linear.sinkhorn_lr) LRSinkhornOutput (class in ott.solvers.linear.sinkhorn_lr) lse_step() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method)

M

make() (ott.geometry.epsilon_scheduler.Epsilon class method) make_kl_loss() (in module ott.problems.quadratic.quadratic_costs) make_rng() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense 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) marginal_dependent_cost() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) marginal_from_potentials() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) marginal_from_scalings() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) mask() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry 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) max_measure_size (ott.problems.linear.barycenter_problem.BarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property)

mean_cost_matrix (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) median_cost_matrix (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) memory (ott.solvers.linear.acceleration.AndersonAcceleration attribute) MetaInitializer (class in ott.initializers.nn.initializers) MetaMLP (class in ott.initializers.nn.initializers) module ott.geometry ott.initializers ott.initializers.linear ott.initializers.nn ott.initializers.quadratic ott.math ott.problems ott.problems.linear ott.problems.quadratic ott.solvers ott.solvers.linear ott.solvers.nn ott.solvers.quadratic ott.tools ott.tools.gaussian_mixture Momentum (class in ott.solvers.linear.acceleration)

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) name (ott.initializers.nn.initializers.MetaMLP attribute) (ott.solvers.nn.icnn.ICNN attribute) (ott.solvers.nn.layers.PositiveDense attribute) ndim (ott.problems.linear.barycenter_problem.BarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) ndim_fused (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) NeuralDualSolver (class in ott.solvers.nn.neuraldual) norm (ott.geometry.costs.Cosine attribute) (ott.geometry.costs.CostFn attribute) (ott.geometry.costs.Euclidean attribute) (ott.geometry.costs.PNormP attribute) (ott.geometry.costs.SqPNorm attribute) (ott.geometry.costs.TICost 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) num_hidden_layers (ott.initializers.nn.initializers.MetaMLP attribute) num_hidden_units (ott.initializers.nn.initializers.MetaMLP attribute) num_measures (ott.problems.linear.barycenter_problem.BarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property)

O

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) ot_prob (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) ott.geometry module ott.initializers module ott.initializers.linear module ott.initializers.nn module ott.initializers.quadratic module ott.math module ott.problems module ott.problems.linear module

ott.problems.quadratic module ott.solvers module ott.solvers.linear module ott.solvers.nn module ott.solvers.quadratic module ott.tools module ott.tools.gaussian_mixture module outer_iterations (ott.solvers.linear.sinkhorn.Sinkhorn property) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn property) output_from_state() (ott.solvers.linear.continuous_barycenter.WassersteinBarycenter method) (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein method) (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter method)

P

pairwise() (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) (ott.geometry.costs.UnbalancedBures method) param() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) parent (ott.initializers.nn.initializers.MetaMLP attribute) (ott.solvers.nn.icnn.ICNN attribute) (ott.solvers.nn.layers.PositiveDense attribute) perturb() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) PNormP (class in ott.geometry.costs) PointCloud (class in ott.geometry.pointcloud)

pos_weights (ott.solvers.nn.icnn.ICNN attribute) PositiveDense (class in ott.solvers.nn.layers) potential_from_scaling() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) potential_size (ott.initializers.nn.initializers.MetaMLP attribute) precision (ott.solvers.nn.layers.PositiveDense attribute) precondition_fun (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) prepare_divergences() (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.pointcloud.PointCloud class method) primal_cost (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) put_variable() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method)

Q

q (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput 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) quantize() (in module ott.tools.soft_sort)

R

r (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) RandomInitializer (class in ott.initializers.linear.initializers_lr) rank (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer property) (ott.initializers.linear.initializers_lr.KMeansInitializer property) (ott.initializers.linear.initializers_lr.RandomInitializer property) (ott.initializers.linear.initializers_lr.Rank2Initializer property) (ott.initializers.quadratic.initializers.LRQuadraticInitializer property) Rank2Initializer (class in ott.initializers.linear.initializers_lr) ranks() (in module ott.tools.soft_sort)

rectifier_fn() (ott.solvers.nn.layers.PositiveDense method) refresh_every (ott.solvers.linear.acceleration.AndersonAcceleration attribute) reg_gw_cost (ott.solvers.quadratic.gromov_wasserstein.GWOutput property) reg_ot_cost (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) rho (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) ridge_identity (ott.solvers.linear.acceleration.AndersonAcceleration attribute) (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) ridge_kernel (ott.solvers.linear.implicit_differentiation.ImplicitDiff 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_epsilon (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) scale_params (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) scaling_from_potential() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) scalings (ott.solvers.linear.sinkhorn.SinkhornOutput property) scope (ott.initializers.nn.initializers.MetaMLP attribute) (ott.solvers.nn.icnn.ICNN attribute) (ott.solvers.nn.layers.PositiveDense 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.BarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) segmented_y_fused (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) set() (ott.solvers.linear.continuous_barycenter.BarycenterState method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState method) set_cost() (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) setup() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) (ott.solvers.nn.neuraldual.NeuralDualSolver method) shape (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry 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) SinkhornInitializer (class in ott.initializers.linear.initializers) SinkhornOutput (class in ott.solvers.linear.sinkhorn) solve() (in module ott.solvers.linear.sinkhorn) (in module ott.solvers.quadratic.gromov_wasserstein) (ott.math.decomposition.DenseCholeskySolver method) (ott.math.decomposition.SparseCholeskySolver method) (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) solver (ott.geometry.graph.Graph property) solver_fun() (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) sort() (in module ott.tools.soft_sort) sort_with() (in module ott.tools.soft_sort) sow() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) SparseCholeskySolver (class in ott.math.decomposition) SqEuclidean (class in ott.geometry.costs) SqPNorm (class in ott.geometry.costs) sqrtm (in module ott.math.matrix_square_root) src_mask (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) start (ott.solvers.linear.acceleration.Momentum attribute) subset() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) symmetric (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute)

T

t (ott.geometry.graph.Graph property) tabulate() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) target (ott.geometry.epsilon_scheduler.Epsilon property) tau (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) tgt_mask (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.pointcloud.PointCloud property) TICost (class in ott.geometry.costs) to_dual_potentials() (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.nn.neuraldual.NeuralDualSolver method) to_low_rank() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) to_LRCGeometry() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method)

to_z() (ott.tools.gaussian_mixture.gaussian.Gaussian method) train_neuraldual() (ott.solvers.nn.neuraldual.NeuralDualSolver 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) transport_from_potentials() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) transport_from_scalings() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) transport_mass (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property)

U

UnbalancedBures (class in ott.geometry.costs) unbind() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method) update() (ott.initializers.nn.initializers.MetaInitializer method) (ott.solvers.linear.acceleration.AndersonAcceleration method) (ott.solvers.linear.continuous_barycenter.BarycenterState 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.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) update_scaling() (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.pointcloud.PointCloud method) update_state() (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter method) use_bias (ott.solvers.nn.layers.PositiveDense attribute)

V

value (ott.solvers.linear.acceleration.Momentum attribute) variable() (ott.initializers.nn.initializers.MetaMLP method) (ott.solvers.nn.icnn.ICNN method) (ott.solvers.nn.layers.PositiveDense method)

variables (ott.initializers.nn.initializers.MetaMLP property) (ott.solvers.nn.icnn.ICNN property) (ott.solvers.nn.layers.PositiveDense property) vec_apply_cost() (ott.geometry.pointcloud.PointCloud method)

W

w2_dist() (ott.tools.gaussian_mixture.gaussian.Gaussian method) warm_start (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein property) WassersteinBarycenter (class in ott.solvers.linear.continuous_barycenter)

WassersteinSolver (class in ott.solvers.was_solver) weight() (ott.solvers.linear.acceleration.Momentum method) weights (ott.problems.linear.barycenter_problem.BarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property)

X

x (ott.solvers.linear.continuous_barycenter.BarycenterState attribute) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute)