logo

Examples

  • Getting Started
  • Tutorials
    • Grid geometry
    • Sinkhorn in all flavors
    • OTT vs. POT
    • Sinkhorn Divergence Hessians
    • Low-Rank Sinkhorn
    • Sinkhorn Barycenters
    • Fitting pairs of coupled GMMs
    • Wasserstein Barycenters of Gaussian Mixture Models
    • Soft Sorting
    • Fairness regularizers
    • Single-cell genomics
    • Gromov-Wasserstein
    • Low-Rank Gromov-Wasserstein
    • Gromov-Wasserstein for multi-omics
    • ICNN Dual Solver
    • Initialization Schemes for Input Convex Neural Network
    • Meta OT and Sinkhorn Initializers

API

  • ott.geometry package
    • 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.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.scale_epsilon
      • 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.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.scale_epsilon
      • 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.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.scale_epsilon
      • 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.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.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.graph
      • ott.geometry.graph.Graph.inv_scale_cost
      • ott.geometry.graph.Graph.is_online
      • ott.geometry.graph.Graph.is_sparse
      • ott.geometry.graph.Graph.is_squared_euclidean
      • ott.geometry.graph.Graph.is_symmetric
      • ott.geometry.graph.Graph.kernel_matrix
      • ott.geometry.graph.Graph.laplacian
      • ott.geometry.graph.Graph.mean_cost_matrix
      • ott.geometry.graph.Graph.median_cost_matrix
      • ott.geometry.graph.Graph.scale_epsilon
      • ott.geometry.graph.Graph.shape
      • ott.geometry.graph.Graph.solver
      • ott.geometry.graph.Graph.src_mask
      • ott.geometry.graph.Graph.t
      • 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.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.scale_epsilon
      • 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.make
      • 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.TICost
      • ott.geometry.costs.TICost.all_pairs
      • ott.geometry.costs.TICost.all_pairs_pairwise
      • ott.geometry.costs.TICost.barycenter
      • ott.geometry.costs.TICost.h
      • ott.geometry.costs.TICost.h_legendre
      • ott.geometry.costs.TICost.pairwise
      • ott.geometry.costs.TICost.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.segment.segment_point_cloud
  • ott.problems package
    • ott.problems.linear package
      • 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.BarycenterProblem
        • ott.problems.linear.barycenter_problem.BarycenterProblem.flattened_b
        • ott.problems.linear.barycenter_problem.BarycenterProblem.flattened_y
        • ott.problems.linear.barycenter_problem.BarycenterProblem.max_measure_size
        • ott.problems.linear.barycenter_problem.BarycenterProblem.ndim
        • ott.problems.linear.barycenter_problem.BarycenterProblem.num_measures
        • ott.problems.linear.barycenter_problem.BarycenterProblem.segmented_y_b
        • ott.problems.linear.barycenter_problem.BarycenterProblem.weights
      • ott.problems.linear.potentials.DualPotentials
        • ott.problems.linear.potentials.DualPotentials.distance
        • 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.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 package
      • 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.solvers package
    • ott.solvers.linear package
      • 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.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
        • 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.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.WassersteinBarycenter
        • ott.solvers.linear.continuous_barycenter.WassersteinBarycenter.init_state
        • ott.solvers.linear.continuous_barycenter.WassersteinBarycenter.output_from_state
        • ott.solvers.linear.continuous_barycenter.WassersteinBarycenter.is_low_rank
      • ott.solvers.linear.continuous_barycenter.BarycenterState
        • ott.solvers.linear.continuous_barycenter.BarycenterState.count
        • ott.solvers.linear.continuous_barycenter.BarycenterState.index
        • ott.solvers.linear.continuous_barycenter.BarycenterState.set
        • ott.solvers.linear.continuous_barycenter.BarycenterState.update
        • ott.solvers.linear.continuous_barycenter.BarycenterState.a
        • ott.solvers.linear.continuous_barycenter.BarycenterState.costs
        • ott.solvers.linear.continuous_barycenter.BarycenterState.errors
        • ott.solvers.linear.continuous_barycenter.BarycenterState.linear_convergence
        • ott.solvers.linear.continuous_barycenter.BarycenterState.x
      • ott.solvers.linear.discrete_barycenter.discrete_barycenter
      • 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.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 package
      • 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.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.errors
        • ott.solvers.quadratic.gw_barycenter.GWBarycenterState.gw_convergence
        • 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 package
      • ott.solvers.nn.neuraldual.NeuralDualSolver
        • ott.solvers.nn.neuraldual.NeuralDualSolver.get_step_fn
        • ott.solvers.nn.neuraldual.NeuralDualSolver.setup
        • ott.solvers.nn.neuraldual.NeuralDualSolver.to_dual_potentials
        • ott.solvers.nn.neuraldual.NeuralDualSolver.train_neuraldual
      • ott.solvers.nn.icnn.ICNN
        • ott.solvers.nn.icnn.ICNN.apply
        • ott.solvers.nn.icnn.ICNN.bind
        • ott.solvers.nn.icnn.ICNN.clone
        • ott.solvers.nn.icnn.ICNN.create_train_state
        • ott.solvers.nn.icnn.ICNN.get_variable
        • ott.solvers.nn.icnn.ICNN.has_rng
        • ott.solvers.nn.icnn.ICNN.has_variable
        • ott.solvers.nn.icnn.ICNN.init
        • ott.solvers.nn.icnn.ICNN.init_fn
        • ott.solvers.nn.icnn.ICNN.init_with_output
        • ott.solvers.nn.icnn.ICNN.is_initializing
        • ott.solvers.nn.icnn.ICNN.is_mutable_collection
        • ott.solvers.nn.icnn.ICNN.make_rng
        • ott.solvers.nn.icnn.ICNN.param
        • ott.solvers.nn.icnn.ICNN.perturb
        • ott.solvers.nn.icnn.ICNN.put_variable
        • ott.solvers.nn.icnn.ICNN.setup
        • ott.solvers.nn.icnn.ICNN.sow
        • ott.solvers.nn.icnn.ICNN.tabulate
        • ott.solvers.nn.icnn.ICNN.unbind
        • ott.solvers.nn.icnn.ICNN.variable
        • ott.solvers.nn.icnn.ICNN.act_fn
        • ott.solvers.nn.icnn.ICNN.gaussian_map
        • ott.solvers.nn.icnn.ICNN.init_std
        • ott.solvers.nn.icnn.ICNN.name
        • ott.solvers.nn.icnn.ICNN.parent
        • ott.solvers.nn.icnn.ICNN.pos_weights
        • ott.solvers.nn.icnn.ICNN.scope
        • ott.solvers.nn.icnn.ICNN.variables
        • ott.solvers.nn.icnn.ICNN.dim_data
        • ott.solvers.nn.icnn.ICNN.dim_hidden
      • ott.solvers.nn.layers.PositiveDense
        • ott.solvers.nn.layers.PositiveDense.apply
        • ott.solvers.nn.layers.PositiveDense.bias_init
        • ott.solvers.nn.layers.PositiveDense.bind
        • ott.solvers.nn.layers.PositiveDense.clone
        • ott.solvers.nn.layers.PositiveDense.get_variable
        • ott.solvers.nn.layers.PositiveDense.has_rng
        • ott.solvers.nn.layers.PositiveDense.has_variable
        • ott.solvers.nn.layers.PositiveDense.init
        • ott.solvers.nn.layers.PositiveDense.init_with_output
        • ott.solvers.nn.layers.PositiveDense.inv_rectifier_fn
        • ott.solvers.nn.layers.PositiveDense.is_initializing
        • ott.solvers.nn.layers.PositiveDense.is_mutable_collection
        • ott.solvers.nn.layers.PositiveDense.kernel_init
        • ott.solvers.nn.layers.PositiveDense.make_rng
        • ott.solvers.nn.layers.PositiveDense.param
        • ott.solvers.nn.layers.PositiveDense.perturb
        • ott.solvers.nn.layers.PositiveDense.put_variable
        • ott.solvers.nn.layers.PositiveDense.rectifier_fn
        • ott.solvers.nn.layers.PositiveDense.setup
        • ott.solvers.nn.layers.PositiveDense.sow
        • ott.solvers.nn.layers.PositiveDense.tabulate
        • ott.solvers.nn.layers.PositiveDense.unbind
        • ott.solvers.nn.layers.PositiveDense.variable
        • ott.solvers.nn.layers.PositiveDense.name
        • ott.solvers.nn.layers.PositiveDense.parent
        • ott.solvers.nn.layers.PositiveDense.precision
        • ott.solvers.nn.layers.PositiveDense.scope
        • ott.solvers.nn.layers.PositiveDense.use_bias
        • ott.solvers.nn.layers.PositiveDense.variables
        • ott.solvers.nn.layers.PositiveDense.dim_hidden
    • ott.solvers.was_solver.WassersteinSolver
      • ott.solvers.was_solver.WassersteinSolver.is_low_rank
  • ott.initializers package
    • ott.initializers.linear package
      • 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.SinkhornInitializer
        • ott.initializers.linear.initializers.SinkhornInitializer.init_dual_a
        • ott.initializers.linear.initializers.SinkhornInitializer.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 package
      • ott.initializers.quadratic.initializers.QuadraticInitializer
      • ott.initializers.quadratic.initializers.LRQuadraticInitializer
        • ott.initializers.quadratic.initializers.LRQuadraticInitializer.rank
    • ott.initializers.nn package
      • 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.bind
        • 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.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 package
    • ott.tools.segment_sinkhorn.segment_sinkhorn
    • 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 package
    • ott.math.fixed_point_loop.fixpoint_iter
    • ott.math.decomposition.DenseCholeskySolver
      • ott.math.decomposition.DenseCholeskySolver.create
      • ott.math.decomposition.DenseCholeskySolver.solve
      • ott.math.decomposition.DenseCholeskySolver.A
      • ott.math.decomposition.DenseCholeskySolver.L
    • ott.math.decomposition.SparseCholeskySolver
      • ott.math.decomposition.SparseCholeskySolver.clear_factor_cache
      • ott.math.decomposition.SparseCholeskySolver.create
      • ott.math.decomposition.SparseCholeskySolver.solve
      • ott.math.decomposition.SparseCholeskySolver.A
      • ott.math.decomposition.SparseCholeskySolver.L
    • ott.math.matrix_square_root.sqrtm

References

  • Bibliography
Theme by the Executable Book Project
  • .rst

ott.initializers package

ott.initializers package#

  • ott.initializers.linear package
    • Sinkhorn Initializers
    • Low-rank Sinkhorn Initializers
  • ott.initializers.quadratic package
    • Gromov-Wasserstein Initializers
  • ott.initializers.nn package
    • Neural Initializers

previous

ott.solvers.was_solver.WassersteinSolver.is_low_rank

next

ott.initializers.linear package

By OTT authors
© Copyright 2021-2023, OTT authors.