Skip to main content
Ctrl
+
K
In 0.4.1, the jit argument in solvers will be removed. Please jit the solvers explicitly.
Examples
Getting Started
Tutorials
Grid Geometry
Point Clouds
Sinkhorn in All Flavors
OTT vs. POT
Sinkhorn Divergence Hessians
Low-rank Sinkhorn
Sinkhorn divergence gradient flows
Sinkhorn Barycenters
Fitting Pairs of Coupled GMMs
GMM Barycenters
Tracking progress of
ott.solvers
Soft Sorting
Single-cell Genomics
Gromov-Wasserstein
Low-Rank GW
GW for Multi-omics
Neural Dual Solver
ICNN Initialization
Meta OT
API
ott.geometry
ott.geometry.geometry.Geometry
ott.geometry.geometry.Geometry.apply_cost
ott.geometry.geometry.Geometry.apply_kernel
ott.geometry.geometry.Geometry.apply_lse_kernel
ott.geometry.geometry.Geometry.apply_square_cost
ott.geometry.geometry.Geometry.apply_transport_from_potentials
ott.geometry.geometry.Geometry.apply_transport_from_scalings
ott.geometry.geometry.Geometry.copy_epsilon
ott.geometry.geometry.Geometry.marginal_from_potentials
ott.geometry.geometry.Geometry.marginal_from_scalings
ott.geometry.geometry.Geometry.mask
ott.geometry.geometry.Geometry.potential_from_scaling
ott.geometry.geometry.Geometry.prepare_divergences
ott.geometry.geometry.Geometry.scaling_from_potential
ott.geometry.geometry.Geometry.set_scale_cost
ott.geometry.geometry.Geometry.subset
ott.geometry.geometry.Geometry.to_LRCGeometry
ott.geometry.geometry.Geometry.transport_from_potentials
ott.geometry.geometry.Geometry.transport_from_scalings
ott.geometry.geometry.Geometry.update_potential
ott.geometry.geometry.Geometry.update_scaling
ott.geometry.geometry.Geometry.can_LRC
ott.geometry.geometry.Geometry.cost_matrix
ott.geometry.geometry.Geometry.cost_rank
ott.geometry.geometry.Geometry.dtype
ott.geometry.geometry.Geometry.epsilon
ott.geometry.geometry.Geometry.inv_scale_cost
ott.geometry.geometry.Geometry.is_online
ott.geometry.geometry.Geometry.is_squared_euclidean
ott.geometry.geometry.Geometry.is_symmetric
ott.geometry.geometry.Geometry.kernel_matrix
ott.geometry.geometry.Geometry.mean_cost_matrix
ott.geometry.geometry.Geometry.median_cost_matrix
ott.geometry.geometry.Geometry.shape
ott.geometry.geometry.Geometry.src_mask
ott.geometry.geometry.Geometry.tgt_mask
ott.geometry.pointcloud.PointCloud
ott.geometry.pointcloud.PointCloud.apply_cost
ott.geometry.pointcloud.PointCloud.apply_kernel
ott.geometry.pointcloud.PointCloud.apply_lse_kernel
ott.geometry.pointcloud.PointCloud.apply_square_cost
ott.geometry.pointcloud.PointCloud.apply_transport_from_potentials
ott.geometry.pointcloud.PointCloud.apply_transport_from_scalings
ott.geometry.pointcloud.PointCloud.barycenter
ott.geometry.pointcloud.PointCloud.copy_epsilon
ott.geometry.pointcloud.PointCloud.marginal_from_potentials
ott.geometry.pointcloud.PointCloud.marginal_from_scalings
ott.geometry.pointcloud.PointCloud.mask
ott.geometry.pointcloud.PointCloud.potential_from_scaling
ott.geometry.pointcloud.PointCloud.prepare_divergences
ott.geometry.pointcloud.PointCloud.scaling_from_potential
ott.geometry.pointcloud.PointCloud.set_scale_cost
ott.geometry.pointcloud.PointCloud.subset
ott.geometry.pointcloud.PointCloud.to_LRCGeometry
ott.geometry.pointcloud.PointCloud.transport_from_potentials
ott.geometry.pointcloud.PointCloud.transport_from_scalings
ott.geometry.pointcloud.PointCloud.update_potential
ott.geometry.pointcloud.PointCloud.update_scaling
ott.geometry.pointcloud.PointCloud.vec_apply_cost
ott.geometry.pointcloud.PointCloud.batch_size
ott.geometry.pointcloud.PointCloud.can_LRC
ott.geometry.pointcloud.PointCloud.cost_matrix
ott.geometry.pointcloud.PointCloud.cost_rank
ott.geometry.pointcloud.PointCloud.dtype
ott.geometry.pointcloud.PointCloud.epsilon
ott.geometry.pointcloud.PointCloud.inv_scale_cost
ott.geometry.pointcloud.PointCloud.is_online
ott.geometry.pointcloud.PointCloud.is_squared_euclidean
ott.geometry.pointcloud.PointCloud.is_symmetric
ott.geometry.pointcloud.PointCloud.kernel_matrix
ott.geometry.pointcloud.PointCloud.mean_cost_matrix
ott.geometry.pointcloud.PointCloud.median_cost_matrix
ott.geometry.pointcloud.PointCloud.shape
ott.geometry.pointcloud.PointCloud.src_mask
ott.geometry.pointcloud.PointCloud.tgt_mask
ott.geometry.grid.Grid
ott.geometry.grid.Grid.apply_cost
ott.geometry.grid.Grid.apply_kernel
ott.geometry.grid.Grid.apply_lse_kernel
ott.geometry.grid.Grid.apply_square_cost
ott.geometry.grid.Grid.apply_transport_from_potentials
ott.geometry.grid.Grid.apply_transport_from_scalings
ott.geometry.grid.Grid.copy_epsilon
ott.geometry.grid.Grid.marginal_from_potentials
ott.geometry.grid.Grid.marginal_from_scalings
ott.geometry.grid.Grid.mask
ott.geometry.grid.Grid.potential_from_scaling
ott.geometry.grid.Grid.prepare_divergences
ott.geometry.grid.Grid.scaling_from_potential
ott.geometry.grid.Grid.set_scale_cost
ott.geometry.grid.Grid.subset
ott.geometry.grid.Grid.to_LRCGeometry
ott.geometry.grid.Grid.transport_from_potentials
ott.geometry.grid.Grid.transport_from_scalings
ott.geometry.grid.Grid.update_potential
ott.geometry.grid.Grid.update_scaling
ott.geometry.grid.Grid.can_LRC
ott.geometry.grid.Grid.cost_matrix
ott.geometry.grid.Grid.cost_rank
ott.geometry.grid.Grid.dtype
ott.geometry.grid.Grid.epsilon
ott.geometry.grid.Grid.geometries
ott.geometry.grid.Grid.inv_scale_cost
ott.geometry.grid.Grid.is_online
ott.geometry.grid.Grid.is_squared_euclidean
ott.geometry.grid.Grid.is_symmetric
ott.geometry.grid.Grid.kernel_matrix
ott.geometry.grid.Grid.mean_cost_matrix
ott.geometry.grid.Grid.median_cost_matrix
ott.geometry.grid.Grid.shape
ott.geometry.grid.Grid.src_mask
ott.geometry.grid.Grid.tgt_mask
ott.geometry.graph.Graph
ott.geometry.graph.Graph.apply_cost
ott.geometry.graph.Graph.apply_kernel
ott.geometry.graph.Graph.apply_lse_kernel
ott.geometry.graph.Graph.apply_square_cost
ott.geometry.graph.Graph.apply_transport_from_potentials
ott.geometry.graph.Graph.apply_transport_from_scalings
ott.geometry.graph.Graph.copy_epsilon
ott.geometry.graph.Graph.from_graph
ott.geometry.graph.Graph.marginal_from_potentials
ott.geometry.graph.Graph.marginal_from_scalings
ott.geometry.graph.Graph.mask
ott.geometry.graph.Graph.potential_from_scaling
ott.geometry.graph.Graph.prepare_divergences
ott.geometry.graph.Graph.scaling_from_potential
ott.geometry.graph.Graph.set_scale_cost
ott.geometry.graph.Graph.subset
ott.geometry.graph.Graph.to_LRCGeometry
ott.geometry.graph.Graph.transport_from_potentials
ott.geometry.graph.Graph.transport_from_scalings
ott.geometry.graph.Graph.update_potential
ott.geometry.graph.Graph.update_scaling
ott.geometry.graph.Graph.can_LRC
ott.geometry.graph.Graph.cost_matrix
ott.geometry.graph.Graph.cost_rank
ott.geometry.graph.Graph.dtype
ott.geometry.graph.Graph.epsilon
ott.geometry.graph.Graph.inv_scale_cost
ott.geometry.graph.Graph.is_online
ott.geometry.graph.Graph.is_squared_euclidean
ott.geometry.graph.Graph.is_symmetric
ott.geometry.graph.Graph.kernel_matrix
ott.geometry.graph.Graph.mean_cost_matrix
ott.geometry.graph.Graph.median_cost_matrix
ott.geometry.graph.Graph.shape
ott.geometry.graph.Graph.src_mask
ott.geometry.graph.Graph.tgt_mask
ott.geometry.low_rank.LRCGeometry
ott.geometry.low_rank.LRCGeometry.apply_cost
ott.geometry.low_rank.LRCGeometry.apply_kernel
ott.geometry.low_rank.LRCGeometry.apply_lse_kernel
ott.geometry.low_rank.LRCGeometry.apply_square_cost
ott.geometry.low_rank.LRCGeometry.apply_transport_from_potentials
ott.geometry.low_rank.LRCGeometry.apply_transport_from_scalings
ott.geometry.low_rank.LRCGeometry.compute_max_cost
ott.geometry.low_rank.LRCGeometry.copy_epsilon
ott.geometry.low_rank.LRCGeometry.marginal_from_potentials
ott.geometry.low_rank.LRCGeometry.marginal_from_scalings
ott.geometry.low_rank.LRCGeometry.mask
ott.geometry.low_rank.LRCGeometry.potential_from_scaling
ott.geometry.low_rank.LRCGeometry.prepare_divergences
ott.geometry.low_rank.LRCGeometry.scaling_from_potential
ott.geometry.low_rank.LRCGeometry.set_scale_cost
ott.geometry.low_rank.LRCGeometry.subset
ott.geometry.low_rank.LRCGeometry.to_LRCGeometry
ott.geometry.low_rank.LRCGeometry.transport_from_potentials
ott.geometry.low_rank.LRCGeometry.transport_from_scalings
ott.geometry.low_rank.LRCGeometry.update_potential
ott.geometry.low_rank.LRCGeometry.update_scaling
ott.geometry.low_rank.LRCGeometry.bias
ott.geometry.low_rank.LRCGeometry.can_LRC
ott.geometry.low_rank.LRCGeometry.cost_1
ott.geometry.low_rank.LRCGeometry.cost_2
ott.geometry.low_rank.LRCGeometry.cost_matrix
ott.geometry.low_rank.LRCGeometry.cost_rank
ott.geometry.low_rank.LRCGeometry.dtype
ott.geometry.low_rank.LRCGeometry.epsilon
ott.geometry.low_rank.LRCGeometry.inv_scale_cost
ott.geometry.low_rank.LRCGeometry.is_online
ott.geometry.low_rank.LRCGeometry.is_squared_euclidean
ott.geometry.low_rank.LRCGeometry.is_symmetric
ott.geometry.low_rank.LRCGeometry.kernel_matrix
ott.geometry.low_rank.LRCGeometry.mean_cost_matrix
ott.geometry.low_rank.LRCGeometry.median_cost_matrix
ott.geometry.low_rank.LRCGeometry.shape
ott.geometry.low_rank.LRCGeometry.src_mask
ott.geometry.low_rank.LRCGeometry.tgt_mask
ott.geometry.epsilon_scheduler.Epsilon
ott.geometry.epsilon_scheduler.Epsilon.at
ott.geometry.epsilon_scheduler.Epsilon.done
ott.geometry.epsilon_scheduler.Epsilon.done_at
ott.geometry.epsilon_scheduler.Epsilon.set
ott.geometry.epsilon_scheduler.Epsilon.target
ott.geometry.costs.CostFn
ott.geometry.costs.CostFn.all_pairs
ott.geometry.costs.CostFn.all_pairs_pairwise
ott.geometry.costs.CostFn.barycenter
ott.geometry.costs.CostFn.pairwise
ott.geometry.costs.CostFn.norm
ott.geometry.costs.SqPNorm
ott.geometry.costs.SqPNorm.all_pairs
ott.geometry.costs.SqPNorm.all_pairs_pairwise
ott.geometry.costs.SqPNorm.barycenter
ott.geometry.costs.SqPNorm.h
ott.geometry.costs.SqPNorm.h_legendre
ott.geometry.costs.SqPNorm.pairwise
ott.geometry.costs.SqPNorm.norm
ott.geometry.costs.PNormP
ott.geometry.costs.PNormP.all_pairs
ott.geometry.costs.PNormP.all_pairs_pairwise
ott.geometry.costs.PNormP.barycenter
ott.geometry.costs.PNormP.h
ott.geometry.costs.PNormP.h_legendre
ott.geometry.costs.PNormP.pairwise
ott.geometry.costs.PNormP.norm
ott.geometry.costs.SqEuclidean
ott.geometry.costs.SqEuclidean.all_pairs
ott.geometry.costs.SqEuclidean.all_pairs_pairwise
ott.geometry.costs.SqEuclidean.barycenter
ott.geometry.costs.SqEuclidean.h
ott.geometry.costs.SqEuclidean.h_legendre
ott.geometry.costs.SqEuclidean.norm
ott.geometry.costs.SqEuclidean.pairwise
ott.geometry.costs.Euclidean
ott.geometry.costs.Euclidean.all_pairs
ott.geometry.costs.Euclidean.all_pairs_pairwise
ott.geometry.costs.Euclidean.barycenter
ott.geometry.costs.Euclidean.pairwise
ott.geometry.costs.Euclidean.norm
ott.geometry.costs.Cosine
ott.geometry.costs.Cosine.all_pairs
ott.geometry.costs.Cosine.all_pairs_pairwise
ott.geometry.costs.Cosine.barycenter
ott.geometry.costs.Cosine.pairwise
ott.geometry.costs.Cosine.norm
ott.geometry.costs.Bures
ott.geometry.costs.Bures.all_pairs
ott.geometry.costs.Bures.all_pairs_pairwise
ott.geometry.costs.Bures.barycenter
ott.geometry.costs.Bures.covariance_fixpoint_iter
ott.geometry.costs.Bures.norm
ott.geometry.costs.Bures.pairwise
ott.geometry.costs.UnbalancedBures
ott.geometry.costs.UnbalancedBures.all_pairs
ott.geometry.costs.UnbalancedBures.all_pairs_pairwise
ott.geometry.costs.UnbalancedBures.barycenter
ott.geometry.costs.UnbalancedBures.norm
ott.geometry.costs.UnbalancedBures.pairwise
ott.geometry.costs.ElasticL1
ott.geometry.costs.ElasticL1.all_pairs
ott.geometry.costs.ElasticL1.all_pairs_pairwise
ott.geometry.costs.ElasticL1.barycenter
ott.geometry.costs.ElasticL1.h
ott.geometry.costs.ElasticL1.h_legendre
ott.geometry.costs.ElasticL1.pairwise
ott.geometry.costs.ElasticL1.prox_reg
ott.geometry.costs.ElasticL1.reg
ott.geometry.costs.ElasticL1.norm
ott.geometry.costs.ElasticSTVS
ott.geometry.costs.ElasticSTVS.all_pairs
ott.geometry.costs.ElasticSTVS.all_pairs_pairwise
ott.geometry.costs.ElasticSTVS.barycenter
ott.geometry.costs.ElasticSTVS.h
ott.geometry.costs.ElasticSTVS.h_legendre
ott.geometry.costs.ElasticSTVS.pairwise
ott.geometry.costs.ElasticSTVS.prox_reg
ott.geometry.costs.ElasticSTVS.reg
ott.geometry.costs.ElasticSTVS.norm
ott.geometry.costs.ElasticSqKOverlap
ott.geometry.costs.ElasticSqKOverlap.all_pairs
ott.geometry.costs.ElasticSqKOverlap.all_pairs_pairwise
ott.geometry.costs.ElasticSqKOverlap.barycenter
ott.geometry.costs.ElasticSqKOverlap.h
ott.geometry.costs.ElasticSqKOverlap.h_legendre
ott.geometry.costs.ElasticSqKOverlap.pairwise
ott.geometry.costs.ElasticSqKOverlap.prox_reg
ott.geometry.costs.ElasticSqKOverlap.reg
ott.geometry.costs.ElasticSqKOverlap.norm
ott.geometry.costs.SoftDTW
ott.geometry.costs.SoftDTW.all_pairs
ott.geometry.costs.SoftDTW.all_pairs_pairwise
ott.geometry.costs.SoftDTW.barycenter
ott.geometry.costs.SoftDTW.pairwise
ott.geometry.costs.SoftDTW.norm
ott.geometry.segment.segment_point_cloud
ott.problems
ott.problems.linear
ott.problems.linear.linear_problem.LinearProblem
ott.problems.linear.linear_problem.LinearProblem.get_transport_functions
ott.problems.linear.linear_problem.LinearProblem.a
ott.problems.linear.linear_problem.LinearProblem.b
ott.problems.linear.linear_problem.LinearProblem.epsilon
ott.problems.linear.linear_problem.LinearProblem.is_balanced
ott.problems.linear.barycenter_problem.FixedBarycenterProblem
ott.problems.linear.barycenter_problem.FixedBarycenterProblem.num_measures
ott.problems.linear.barycenter_problem.FixedBarycenterProblem.weights
ott.problems.linear.barycenter_problem.FreeBarycenterProblem
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.flattened_b
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.flattened_y
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.max_measure_size
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.ndim
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.num_measures
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.segmented_y_b
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.weights
ott.problems.linear.potentials.DualPotentials
ott.problems.linear.potentials.DualPotentials.distance
ott.problems.linear.potentials.DualPotentials.plot_ot_map
ott.problems.linear.potentials.DualPotentials.plot_potential
ott.problems.linear.potentials.DualPotentials.transport
ott.problems.linear.potentials.DualPotentials.f
ott.problems.linear.potentials.DualPotentials.g
ott.problems.linear.potentials.EntropicPotentials
ott.problems.linear.potentials.EntropicPotentials.distance
ott.problems.linear.potentials.EntropicPotentials.plot_ot_map
ott.problems.linear.potentials.EntropicPotentials.plot_potential
ott.problems.linear.potentials.EntropicPotentials.transport
ott.problems.linear.potentials.EntropicPotentials.epsilon
ott.problems.linear.potentials.EntropicPotentials.f
ott.problems.linear.potentials.EntropicPotentials.g
ott.problems.linear.potentials.EntropicPotentials.is_debiased
ott.problems.quadratic
ott.problems.quadratic.quadratic_problem.QuadraticProblem
ott.problems.quadratic.quadratic_problem.QuadraticProblem.cost_unbalanced_correction
ott.problems.quadratic.quadratic_problem.QuadraticProblem.init_transport_mass
ott.problems.quadratic.quadratic_problem.QuadraticProblem.marginal_dependent_cost
ott.problems.quadratic.quadratic_problem.QuadraticProblem.to_low_rank
ott.problems.quadratic.quadratic_problem.QuadraticProblem.update_linearization
ott.problems.quadratic.quadratic_problem.QuadraticProblem.update_lr_geom
ott.problems.quadratic.quadratic_problem.QuadraticProblem.update_lr_linearization
ott.problems.quadratic.quadratic_problem.QuadraticProblem.a
ott.problems.quadratic.quadratic_problem.QuadraticProblem.b
ott.problems.quadratic.quadratic_problem.QuadraticProblem.geom_xx
ott.problems.quadratic.quadratic_problem.QuadraticProblem.geom_xy
ott.problems.quadratic.quadratic_problem.QuadraticProblem.geom_yy
ott.problems.quadratic.quadratic_problem.QuadraticProblem.is_balanced
ott.problems.quadratic.quadratic_problem.QuadraticProblem.is_fused
ott.problems.quadratic.quadratic_problem.QuadraticProblem.is_low_rank
ott.problems.quadratic.quadratic_problem.QuadraticProblem.linear_loss
ott.problems.quadratic.quadratic_problem.QuadraticProblem.quad_loss
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.update_barycenter
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.update_features
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.flattened_b
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.flattened_y
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.gw_loss
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.is_fused
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.max_measure_size
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.ndim
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.ndim_fused
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.num_measures
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.segmented_y_b
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.segmented_y_fused
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.weights
ott.problems.quadratic.quadratic_costs.make_square_loss
ott.problems.quadratic.quadratic_costs.make_kl_loss
ott.problems.nn
ott.problems.nn.dataset.create_gaussian_mixture_samplers
ott.problems.nn.dataset.Dataset
ott.problems.nn.dataset.Dataset.count
ott.problems.nn.dataset.Dataset.index
ott.problems.nn.dataset.Dataset.source_iter
ott.problems.nn.dataset.Dataset.target_iter
ott.problems.nn.dataset.GaussianMixture
ott.problems.nn.dataset.GaussianMixture.scale
ott.problems.nn.dataset.GaussianMixture.variance
ott.problems.nn.dataset.GaussianMixture.name
ott.problems.nn.dataset.GaussianMixture.batch_size
ott.problems.nn.dataset.GaussianMixture.init_rng
ott.solvers
ott.solvers.linear
ott.solvers.linear.sinkhorn.solve
ott.solvers.linear.sinkhorn.Sinkhorn
ott.solvers.linear.sinkhorn.Sinkhorn.create_initializer
ott.solvers.linear.sinkhorn.Sinkhorn.init_state
ott.solvers.linear.sinkhorn.Sinkhorn.kernel_step
ott.solvers.linear.sinkhorn.Sinkhorn.lse_step
ott.solvers.linear.sinkhorn.Sinkhorn.one_iteration
ott.solvers.linear.sinkhorn.Sinkhorn.output_from_state
ott.solvers.linear.sinkhorn.Sinkhorn.norm_error
ott.solvers.linear.sinkhorn.Sinkhorn.outer_iterations
ott.solvers.linear.sinkhorn.SinkhornState
ott.solvers.linear.sinkhorn.SinkhornState.count
ott.solvers.linear.sinkhorn.SinkhornState.ent_reg_cost
ott.solvers.linear.sinkhorn.SinkhornState.index
ott.solvers.linear.sinkhorn.SinkhornState.recenter
ott.solvers.linear.sinkhorn.SinkhornState.set
ott.solvers.linear.sinkhorn.SinkhornState.solution_error
ott.solvers.linear.sinkhorn.SinkhornState.errors
ott.solvers.linear.sinkhorn.SinkhornState.fu
ott.solvers.linear.sinkhorn.SinkhornState.gv
ott.solvers.linear.sinkhorn.SinkhornState.old_fus
ott.solvers.linear.sinkhorn.SinkhornState.old_mapped_fus
ott.solvers.linear.sinkhorn.SinkhornOutput
ott.solvers.linear.sinkhorn.SinkhornOutput.apply
ott.solvers.linear.sinkhorn.SinkhornOutput.cost_at_geom
ott.solvers.linear.sinkhorn.SinkhornOutput.count
ott.solvers.linear.sinkhorn.SinkhornOutput.index
ott.solvers.linear.sinkhorn.SinkhornOutput.marginal
ott.solvers.linear.sinkhorn.SinkhornOutput.set
ott.solvers.linear.sinkhorn.SinkhornOutput.set_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.to_dual_potentials
ott.solvers.linear.sinkhorn.SinkhornOutput.transport_cost_at_geom
ott.solvers.linear.sinkhorn.SinkhornOutput.a
ott.solvers.linear.sinkhorn.SinkhornOutput.b
ott.solvers.linear.sinkhorn.SinkhornOutput.converged
ott.solvers.linear.sinkhorn.SinkhornOutput.dual_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.errors
ott.solvers.linear.sinkhorn.SinkhornOutput.f
ott.solvers.linear.sinkhorn.SinkhornOutput.g
ott.solvers.linear.sinkhorn.SinkhornOutput.geom
ott.solvers.linear.sinkhorn.SinkhornOutput.linear
ott.solvers.linear.sinkhorn.SinkhornOutput.linear_output
ott.solvers.linear.sinkhorn.SinkhornOutput.matrix
ott.solvers.linear.sinkhorn.SinkhornOutput.n_iters
ott.solvers.linear.sinkhorn.SinkhornOutput.ot_prob
ott.solvers.linear.sinkhorn.SinkhornOutput.primal_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.reg_ot_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.scalings
ott.solvers.linear.sinkhorn.SinkhornOutput.transport_mass
ott.solvers.linear.sinkhorn_lr.LRSinkhorn
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.create_initializer
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.dykstra_update_kernel
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.dykstra_update_lse
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.init_state
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.kernel_step
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.lse_step
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.one_iteration
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.output_from_state
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.is_entropic
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.norm_error
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.outer_iterations
ott.solvers.linear.sinkhorn_lr.LRSinkhornState
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.compute_error
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.count
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.index
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.reg_ot_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.set
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.solution_error
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.costs
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.crossed_threshold
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.errors
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.g
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.gamma
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.q
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.r
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.apply
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.compute_reg_ot_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.cost_at_geom
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.count
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.index
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.marginal
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.set
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.set_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.transport_cost_at_geom
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.a
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.b
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.converged
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.costs
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.errors
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.g
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.geom
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.linear
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.linear_output
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.matrix
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.ot_prob
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.primal_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.q
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.r
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.reg_ot_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.transport_mass
ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter
ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter.init_state
ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter.output_from_state
ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter.is_low_rank
ott.solvers.linear.continuous_barycenter.FreeBarycenterState
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.count
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.index
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.set
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.update
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.a
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.costs
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.errors
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.linear_convergence
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.x
ott.solvers.linear.discrete_barycenter.FixedBarycenter
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.count
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.index
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.errors
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.f
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.g
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.histogram
ott.solvers.linear.acceleration.Momentum
ott.solvers.linear.acceleration.Momentum.lehmann
ott.solvers.linear.acceleration.Momentum.weight
ott.solvers.linear.acceleration.Momentum.error_threshold
ott.solvers.linear.acceleration.Momentum.inner_iterations
ott.solvers.linear.acceleration.Momentum.start
ott.solvers.linear.acceleration.Momentum.value
ott.solvers.linear.acceleration.AndersonAcceleration
ott.solvers.linear.acceleration.AndersonAcceleration.extrapolation
ott.solvers.linear.acceleration.AndersonAcceleration.init_maps
ott.solvers.linear.acceleration.AndersonAcceleration.update
ott.solvers.linear.acceleration.AndersonAcceleration.update_history
ott.solvers.linear.acceleration.AndersonAcceleration.memory
ott.solvers.linear.acceleration.AndersonAcceleration.refresh_every
ott.solvers.linear.acceleration.AndersonAcceleration.ridge_identity
ott.solvers.linear.implicit_differentiation.ImplicitDiff
ott.solvers.linear.implicit_differentiation.ImplicitDiff.first_order_conditions
ott.solvers.linear.implicit_differentiation.ImplicitDiff.gradient
ott.solvers.linear.implicit_differentiation.ImplicitDiff.replace
ott.solvers.linear.implicit_differentiation.ImplicitDiff.solve
ott.solvers.linear.implicit_differentiation.ImplicitDiff.solver_fun
ott.solvers.linear.implicit_differentiation.ImplicitDiff.precondition_fun
ott.solvers.linear.implicit_differentiation.ImplicitDiff.ridge_identity
ott.solvers.linear.implicit_differentiation.ImplicitDiff.ridge_kernel
ott.solvers.linear.implicit_differentiation.ImplicitDiff.symmetric
ott.solvers.quadratic
ott.solvers.quadratic.gromov_wasserstein.solve
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.create_initializer
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.init_state
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.output_from_state
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.is_low_rank
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.warm_start
ott.solvers.quadratic.gromov_wasserstein.GWOutput
ott.solvers.quadratic.gromov_wasserstein.GWOutput.apply
ott.solvers.quadratic.gromov_wasserstein.GWOutput.count
ott.solvers.quadratic.gromov_wasserstein.GWOutput.index
ott.solvers.quadratic.gromov_wasserstein.GWOutput.set
ott.solvers.quadratic.gromov_wasserstein.GWOutput.converged
ott.solvers.quadratic.gromov_wasserstein.GWOutput.costs
ott.solvers.quadratic.gromov_wasserstein.GWOutput.errors
ott.solvers.quadratic.gromov_wasserstein.GWOutput.geom
ott.solvers.quadratic.gromov_wasserstein.GWOutput.linear_convergence
ott.solvers.quadratic.gromov_wasserstein.GWOutput.linear_state
ott.solvers.quadratic.gromov_wasserstein.GWOutput.matrix
ott.solvers.quadratic.gromov_wasserstein.GWOutput.old_transport_mass
ott.solvers.quadratic.gromov_wasserstein.GWOutput.primal_cost
ott.solvers.quadratic.gromov_wasserstein.GWOutput.reg_gw_cost
ott.solvers.quadratic.gw_barycenter.GWBarycenterState
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.count
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.index
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.set
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.a
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.cost
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.costs
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.costs_bary
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.errors
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.gw_convergence
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.n_iters
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.x
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.init_state
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.output_from_state
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.update_state
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.is_low_rank
ott.solvers.nn
ott.solvers.nn.neuraldual.W2NeuralDual
ott.solvers.nn.neuraldual.W2NeuralDual.get_step_fn
ott.solvers.nn.neuraldual.W2NeuralDual.setup
ott.solvers.nn.neuraldual.W2NeuralDual.to_dual_potentials
ott.solvers.nn.neuraldual.W2NeuralDual.train_neuraldual_alternating
ott.solvers.nn.neuraldual.W2NeuralDual.train_neuraldual_parallel
ott.solvers.nn.models.ModelBase
ott.solvers.nn.models.ModelBase.apply
ott.solvers.nn.models.ModelBase.clone
ott.solvers.nn.models.ModelBase.create_train_state
ott.solvers.nn.models.ModelBase.get_variable
ott.solvers.nn.models.ModelBase.has_rng
ott.solvers.nn.models.ModelBase.has_variable
ott.solvers.nn.models.ModelBase.init
ott.solvers.nn.models.ModelBase.init_with_output
ott.solvers.nn.models.ModelBase.is_initializing
ott.solvers.nn.models.ModelBase.is_mutable_collection
ott.solvers.nn.models.ModelBase.lazy_init
ott.solvers.nn.models.ModelBase.make_rng
ott.solvers.nn.models.ModelBase.param
ott.solvers.nn.models.ModelBase.perturb
ott.solvers.nn.models.ModelBase.potential_gradient_fn
ott.solvers.nn.models.ModelBase.potential_value_fn
ott.solvers.nn.models.ModelBase.put_variable
ott.solvers.nn.models.ModelBase.setup
ott.solvers.nn.models.ModelBase.sow
ott.solvers.nn.models.ModelBase.tabulate
ott.solvers.nn.models.ModelBase.unbind
ott.solvers.nn.models.ModelBase.variable
ott.solvers.nn.models.ModelBase.is_potential
ott.solvers.nn.models.ModelBase.name
ott.solvers.nn.models.ModelBase.parent
ott.solvers.nn.models.ModelBase.scope
ott.solvers.nn.models.ModelBase.variables
ott.solvers.nn.models.ICNN
ott.solvers.nn.models.ICNN.apply
ott.solvers.nn.models.ICNN.clone
ott.solvers.nn.models.ICNN.create_train_state
ott.solvers.nn.models.ICNN.get_variable
ott.solvers.nn.models.ICNN.has_rng
ott.solvers.nn.models.ICNN.has_variable
ott.solvers.nn.models.ICNN.init
ott.solvers.nn.models.ICNN.init_fn
ott.solvers.nn.models.ICNN.init_with_output
ott.solvers.nn.models.ICNN.is_initializing
ott.solvers.nn.models.ICNN.is_mutable_collection
ott.solvers.nn.models.ICNN.lazy_init
ott.solvers.nn.models.ICNN.make_rng
ott.solvers.nn.models.ICNN.param
ott.solvers.nn.models.ICNN.perturb
ott.solvers.nn.models.ICNN.potential_gradient_fn
ott.solvers.nn.models.ICNN.potential_value_fn
ott.solvers.nn.models.ICNN.put_variable
ott.solvers.nn.models.ICNN.setup
ott.solvers.nn.models.ICNN.sow
ott.solvers.nn.models.ICNN.tabulate
ott.solvers.nn.models.ICNN.unbind
ott.solvers.nn.models.ICNN.variable
ott.solvers.nn.models.ICNN.act_fn
ott.solvers.nn.models.ICNN.gaussian_map
ott.solvers.nn.models.ICNN.init_std
ott.solvers.nn.models.ICNN.is_potential
ott.solvers.nn.models.ICNN.name
ott.solvers.nn.models.ICNN.parent
ott.solvers.nn.models.ICNN.pos_weights
ott.solvers.nn.models.ICNN.scope
ott.solvers.nn.models.ICNN.variables
ott.solvers.nn.models.ICNN.dim_data
ott.solvers.nn.models.ICNN.dim_hidden
ott.solvers.nn.models.MLP
ott.solvers.nn.models.MLP.act_fn
ott.solvers.nn.models.MLP.apply
ott.solvers.nn.models.MLP.clone
ott.solvers.nn.models.MLP.create_train_state
ott.solvers.nn.models.MLP.get_variable
ott.solvers.nn.models.MLP.has_rng
ott.solvers.nn.models.MLP.has_variable
ott.solvers.nn.models.MLP.init
ott.solvers.nn.models.MLP.init_with_output
ott.solvers.nn.models.MLP.is_initializing
ott.solvers.nn.models.MLP.is_mutable_collection
ott.solvers.nn.models.MLP.lazy_init
ott.solvers.nn.models.MLP.make_rng
ott.solvers.nn.models.MLP.param
ott.solvers.nn.models.MLP.perturb
ott.solvers.nn.models.MLP.potential_gradient_fn
ott.solvers.nn.models.MLP.potential_value_fn
ott.solvers.nn.models.MLP.put_variable
ott.solvers.nn.models.MLP.setup
ott.solvers.nn.models.MLP.sow
ott.solvers.nn.models.MLP.tabulate
ott.solvers.nn.models.MLP.unbind
ott.solvers.nn.models.MLP.variable
ott.solvers.nn.models.MLP.is_potential
ott.solvers.nn.models.MLP.name
ott.solvers.nn.models.MLP.parent
ott.solvers.nn.models.MLP.scope
ott.solvers.nn.models.MLP.variables
ott.solvers.nn.models.MLP.dim_hidden
ott.solvers.nn.conjugate_solvers.ConjugateResults
ott.solvers.nn.conjugate_solvers.ConjugateResults.count
ott.solvers.nn.conjugate_solvers.ConjugateResults.index
ott.solvers.nn.conjugate_solvers.ConjugateResults.grad
ott.solvers.nn.conjugate_solvers.ConjugateResults.num_iter
ott.solvers.nn.conjugate_solvers.ConjugateResults.val
ott.solvers.nn.conjugate_solvers.FenchelConjugateSolver
ott.solvers.nn.conjugate_solvers.FenchelConjugateSolver.solve
ott.solvers.nn.conjugate_solvers.FenchelConjugateLBFGS
ott.solvers.nn.conjugate_solvers.FenchelConjugateLBFGS.solve
ott.solvers.nn.conjugate_solvers.FenchelConjugateLBFGS.decrease_factor
ott.solvers.nn.conjugate_solvers.FenchelConjugateLBFGS.gtol
ott.solvers.nn.conjugate_solvers.FenchelConjugateLBFGS.linesearch_type
ott.solvers.nn.conjugate_solvers.FenchelConjugateLBFGS.ls_method
ott.solvers.nn.conjugate_solvers.FenchelConjugateLBFGS.max_iter
ott.solvers.nn.conjugate_solvers.FenchelConjugateLBFGS.max_linesearch_iter
ott.solvers.was_solver.WassersteinSolver
ott.solvers.was_solver.WassersteinSolver.is_low_rank
ott.initializers
ott.initializers.linear
ott.initializers.linear.initializers.DefaultInitializer
ott.initializers.linear.initializers.DefaultInitializer.init_dual_a
ott.initializers.linear.initializers.DefaultInitializer.init_dual_b
ott.initializers.linear.initializers.GaussianInitializer
ott.initializers.linear.initializers.GaussianInitializer.init_dual_a
ott.initializers.linear.initializers.GaussianInitializer.init_dual_b
ott.initializers.linear.initializers.SortingInitializer
ott.initializers.linear.initializers.SortingInitializer.init_dual_a
ott.initializers.linear.initializers.SortingInitializer.init_dual_b
ott.initializers.linear.initializers.SubsampleInitializer
ott.initializers.linear.initializers.SubsampleInitializer.init_dual_a
ott.initializers.linear.initializers.SubsampleInitializer.init_dual_b
ott.initializers.linear.initializers_lr.RandomInitializer
ott.initializers.linear.initializers_lr.RandomInitializer.from_solver
ott.initializers.linear.initializers_lr.RandomInitializer.init_g
ott.initializers.linear.initializers_lr.RandomInitializer.init_q
ott.initializers.linear.initializers_lr.RandomInitializer.init_r
ott.initializers.linear.initializers_lr.RandomInitializer.rank
ott.initializers.linear.initializers_lr.Rank2Initializer
ott.initializers.linear.initializers_lr.Rank2Initializer.from_solver
ott.initializers.linear.initializers_lr.Rank2Initializer.init_g
ott.initializers.linear.initializers_lr.Rank2Initializer.init_q
ott.initializers.linear.initializers_lr.Rank2Initializer.init_r
ott.initializers.linear.initializers_lr.Rank2Initializer.rank
ott.initializers.linear.initializers_lr.KMeansInitializer
ott.initializers.linear.initializers_lr.KMeansInitializer.from_solver
ott.initializers.linear.initializers_lr.KMeansInitializer.init_g
ott.initializers.linear.initializers_lr.KMeansInitializer.init_q
ott.initializers.linear.initializers_lr.KMeansInitializer.init_r
ott.initializers.linear.initializers_lr.KMeansInitializer.rank
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.from_solver
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.init_g
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.init_q
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.init_r
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.rank
ott.initializers.quadratic
ott.initializers.quadratic.initializers.QuadraticInitializer
ott.initializers.quadratic.initializers.LRQuadraticInitializer
ott.initializers.quadratic.initializers.LRQuadraticInitializer.rank
ott.initializers.nn
ott.initializers.nn.initializers.MetaInitializer
ott.initializers.nn.initializers.MetaInitializer.init_dual_a
ott.initializers.nn.initializers.MetaInitializer.init_dual_b
ott.initializers.nn.initializers.MetaInitializer.update
ott.initializers.nn.initializers.MetaMLP
ott.initializers.nn.initializers.MetaMLP.apply
ott.initializers.nn.initializers.MetaMLP.clone
ott.initializers.nn.initializers.MetaMLP.get_variable
ott.initializers.nn.initializers.MetaMLP.has_rng
ott.initializers.nn.initializers.MetaMLP.has_variable
ott.initializers.nn.initializers.MetaMLP.init
ott.initializers.nn.initializers.MetaMLP.init_with_output
ott.initializers.nn.initializers.MetaMLP.is_initializing
ott.initializers.nn.initializers.MetaMLP.is_mutable_collection
ott.initializers.nn.initializers.MetaMLP.lazy_init
ott.initializers.nn.initializers.MetaMLP.make_rng
ott.initializers.nn.initializers.MetaMLP.param
ott.initializers.nn.initializers.MetaMLP.perturb
ott.initializers.nn.initializers.MetaMLP.put_variable
ott.initializers.nn.initializers.MetaMLP.setup
ott.initializers.nn.initializers.MetaMLP.sow
ott.initializers.nn.initializers.MetaMLP.tabulate
ott.initializers.nn.initializers.MetaMLP.unbind
ott.initializers.nn.initializers.MetaMLP.variable
ott.initializers.nn.initializers.MetaMLP.name
ott.initializers.nn.initializers.MetaMLP.num_hidden_layers
ott.initializers.nn.initializers.MetaMLP.num_hidden_units
ott.initializers.nn.initializers.MetaMLP.parent
ott.initializers.nn.initializers.MetaMLP.scope
ott.initializers.nn.initializers.MetaMLP.variables
ott.initializers.nn.initializers.MetaMLP.potential_size
ott.tools
ott.tools.segment_sinkhorn.segment_sinkhorn
ott.tools.plot.Plot
ott.tools.plot.Plot.animate
ott.tools.plot.Plot.update
ott.tools.sinkhorn_divergence.sinkhorn_divergence
ott.tools.sinkhorn_divergence.segment_sinkhorn_divergence
ott.tools.soft_sort.quantile
ott.tools.soft_sort.quantile_normalization
ott.tools.soft_sort.quantize
ott.tools.soft_sort.ranks
ott.tools.soft_sort.sort
ott.tools.soft_sort.sort_with
ott.tools.k_means.k_means
ott.tools.k_means.KMeansOutput
ott.tools.k_means.KMeansOutput.count
ott.tools.k_means.KMeansOutput.index
ott.tools.k_means.KMeansOutput.assignment
ott.tools.k_means.KMeansOutput.centroids
ott.tools.k_means.KMeansOutput.converged
ott.tools.k_means.KMeansOutput.error
ott.tools.k_means.KMeansOutput.inner_errors
ott.tools.k_means.KMeansOutput.iteration
ott.tools.gaussian_mixture.gaussian.Gaussian
ott.tools.gaussian_mixture.gaussian.Gaussian.covariance
ott.tools.gaussian_mixture.gaussian.Gaussian.f_potential
ott.tools.gaussian_mixture.gaussian.Gaussian.from_mean_and_cov
ott.tools.gaussian_mixture.gaussian.Gaussian.from_random
ott.tools.gaussian_mixture.gaussian.Gaussian.from_samples
ott.tools.gaussian_mixture.gaussian.Gaussian.from_z
ott.tools.gaussian_mixture.gaussian.Gaussian.log_prob
ott.tools.gaussian_mixture.gaussian.Gaussian.sample
ott.tools.gaussian_mixture.gaussian.Gaussian.to_z
ott.tools.gaussian_mixture.gaussian.Gaussian.transport
ott.tools.gaussian_mixture.gaussian.Gaussian.w2_dist
ott.tools.gaussian_mixture.gaussian.Gaussian.loc
ott.tools.gaussian_mixture.gaussian.Gaussian.n_dimensions
ott.tools.gaussian_mixture.gaussian.Gaussian.scale
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.components
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.conditional_log_prob
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.from_mean_cov_component_weights
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.from_points_and_assignment_probs
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.from_random
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.get_component
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.get_log_component_posterior
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.has_nans
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.log_component_weights
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.log_prob
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.sample
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.cholesky
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.component_weight_ob
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.component_weights
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.covariance
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.dtype
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.loc
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.n_components
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.n_dimensions
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.scale_params
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_bures_geometry
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_cost_matrix
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_normalized_sinkhorn_coupling
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_sinkhorn
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.dtype
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.epsilon
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.gmm0
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.gmm1
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.lock_gmm1
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.rho
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.tau
ott.tools.gaussian_mixture.fit_gmm.initialize
ott.tools.gaussian_mixture.fit_gmm.fit_model_em
ott.tools.gaussian_mixture.fit_gmm_pair.get_fit_model_em_fn
ott.math
ott.math.fixed_point_loop.fixpoint_iter
ott.math.matrix_square_root.sqrtm
ott.utils
ott.utils.default_progress_fn
References
Bibliography
.rst
.pdf
ott.problems.nn.dataset.GaussianMixture.init_rng
Contents
GaussianMixture.init_rng
ott.problems.nn.dataset.GaussianMixture.init_rng
#
GaussianMixture.
init_rng
:
PRNGKeyArray
#
Contents
GaussianMixture.init_rng
