Index

A

a_to_b (ot.utils.OTResult property) abs() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] allclose() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] any() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] arange() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] argmax() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

argmin() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] argsort() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] array_equal() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] assert_same_dtype_device() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] atan2() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] averaged_sgd_entropic_transport() (in module ot.stochastic), [1]

B

b_to_a (ot.utils.OTResult property) Backend (class in ot.backend), [1] barycenter() (in module ot) (in module ot.bregman), [1] (in module ot.lp) barycenter_debiased() (in module ot.bregman), [1] barycenter_sinkhorn() (in module ot.bregman), [1] barycenter_stabilized() (in module ot.bregman), [1] barycenter_unbalanced() (in module ot) (in module ot.unbalanced), [1] barycenter_unbalanced_sinkhorn() (in module ot.unbalanced), [1] barycenter_unbalanced_stabilized() (in module ot.unbalanced), [1]

BaseEstimator (class in ot.utils), [1] BaseTransport (class in ot.da), [1] batch_grad_dual() (in module ot.stochastic), [1] binary_search_circle() (in module ot) (in module ot.lp) bitsize() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] bures_wasserstein_distance() (in module ot.gaussian), [1] bures_wasserstein_mapping() (in module ot.gaussian), [1]

C

c_transform_entropic() (in module ot.stochastic), [1] center_ot_dual() (in module ot.lp) cg() (in module ot.optim), [1] check_number_threads() (in module ot.lp) check_params() (in module ot.utils), [1] check_random_state() (in module ot.utils), [1] citation (ot.utils.OTResult property) clean_zeros() (in module ot.utils), [1] clip() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] co_optimal_transport() (in module ot.coot), [1] co_optimal_transport2() (in module ot.coot), [1] complement_schur() (in module ot.regpath), [1] compute_next_removal() (in module ot.regpath), [1] compute_transport_plan() (in module ot.regpath), [1] concatenate() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] construct_augmented_H() (in module ot.regpath), [1] convolutional_barycenter2d() (in module ot.bregman), [1] convolutional_barycenter2d_debiased() (in module ot.bregman), [1]

coo_matrix() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] coordinate_grad_semi_dual() (in module ot.stochastic), [1] copy() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] cost_normalization() (in module ot.utils), [1] coupling_ (ot.da.EMDLaplaceTransport attribute), [1] (ot.da.EMDTransport attribute), [1] (ot.da.JCPOTTransport attribute), [1] (ot.da.MappingTransport attribute), [1] (ot.da.SinkhornL1l2Transport attribute), [1] (ot.da.SinkhornLpl1Transport attribute), [1] (ot.da.SinkhornTransport attribute), [1] (ot.da.UnbalancedSinkhornTransport attribute), [1] cumsum() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] CupyBackend (class in ot.backend), [1]

D

delta_Omega() (ot.smooth.NegEntropy method), [1] (ot.smooth.Regularization method), [1] (ot.smooth.SparsityConstrained method), [1] (ot.smooth.SquaredL2 method), [1] deprecated (class in ot.utils), [1] detach() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] device_type() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] diag() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

dist() (in module ot) (in module ot.dr), [1] (in module ot.utils), [1] dist0() (in module ot.utils), [1] distribution_estimation_uniform() (in module ot.da), [1] dmmot_monge_1dgrid_loss() (in module ot.lp) dmmot_monge_1dgrid_optimize() (in module ot.lp) dot() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] dots() (in module ot.utils), [1] dtype_device() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] dual_obj_grad() (in module ot.smooth), [1]

E

eigh() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] einsum() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] eliminate_zeros() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] emd() (in module ot) (in module ot.lp), [1] emd2() (in module ot) (in module ot.lp), [1] emd2_1d() (in module ot) (in module ot.lp) emd_1d() (in module ot) (in module ot.lp) emd_laplace() (in module ot.da), [1] EMDLaplaceTransport (class in ot.da), [1] EMDTransport (class in ot.da), [1] empirical_bures_wasserstein_distance() (in module ot.gaussian), [1] empirical_bures_wasserstein_mapping() (in module ot.gaussian), [1] empirical_gaussian_gromov_wasserstein_distance() (in module ot.gaussian), [1]

empirical_gaussian_gromov_wasserstein_mapping() (in module ot.gaussian), [1] empirical_sinkhorn() (in module ot.bregman), [1] empirical_sinkhorn2() (in module ot.bregman), [1] empirical_sinkhorn_divergence() (in module ot.bregman), [1] entropic_fused_gromov_barycenters() (in module ot.gromov) entropic_fused_gromov_wasserstein() (in module ot.gromov) entropic_fused_gromov_wasserstein2() (in module ot.gromov) entropic_gromov_barycenters() (in module ot.gromov) entropic_gromov_wasserstein() (in module ot.gromov) entropic_gromov_wasserstein2() (in module ot.gromov) entropic_partial_gromov_wasserstein() (in module ot.partial), [1] entropic_partial_gromov_wasserstein2() (in module ot.partial), [1] entropic_partial_wasserstein() (in module ot.partial), [1] entropic_semirelaxed_fused_gromov_wasserstein() (in module ot.gromov) entropic_semirelaxed_fused_gromov_wasserstein2() (in module ot.gromov) entropic_semirelaxed_gromov_wasserstein() (in module ot.gromov) entropic_semirelaxed_gromov_wasserstein2() (in module ot.gromov) estimate_dual_null_weights() (in module ot.lp) euclidean_distances() (in module ot.utils), [1] ewca() (in module ot.dr), [1] exp() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] eye() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

F

factored_optimal_transport() (in module ot) (in module ot.factored), [1] fda() (in module ot.dr), [1] fgw_barycenters() (in module ot.gromov) FGW_distance_to_templates() (in module ot.gnn) fit() (ot.da.BaseTransport method), [1] (ot.da.EMDLaplaceTransport method), [1] (ot.da.EMDTransport method), [1] (ot.da.JCPOTTransport method), [1] (ot.da.LinearGWTransport method), [1] (ot.da.LinearTransport method), [1] (ot.da.MappingTransport method), [1] (ot.da.SinkhornL1l2Transport method), [1] (ot.da.SinkhornLpl1Transport method), [1] (ot.da.SinkhornTransport method), [1] (ot.da.UnbalancedSinkhornTransport method), [1] fit_transform() (ot.da.BaseTransport method), [1] flip() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

floor() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] forward() (ot.gnn.TFGWPooling method) (ot.gnn.TWPooling method) free_support_barycenter() (in module ot.lp), [1] free_support_sinkhorn_barycenter() (in module ot.bregman), [1] from_numpy() (ot.backend.Backend method), [1] full() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] fully_relaxed_path() (in module ot.regpath), [1] fused_gromov_wasserstein() (in module ot) (in module ot.gromov) fused_gromov_wasserstein2() (in module ot) (in module ot.gromov) fused_gromov_wasserstein_dictionary_learning() (in module ot.gromov) fused_gromov_wasserstein_linear_unmixing() (in module ot.gromov)

G

gaussian_gromov_wasserstein_distance() (in module ot.gaussian), [1] gaussian_gromov_wasserstein_mapping() (in module ot.gaussian), [1] gcg() (in module ot.optim), [1] generalized_free_support_barycenter() (in module ot.lp), [1] generic_conditional_gradient() (in module ot.optim), [1] geometricBar() (in module ot.bregman), [1] geometricMean() (in module ot.bregman), [1] get_backend() (in module ot.backend), [1] get_backend_list() (in module ot.backend), [1] get_coordinate_circle() (in module ot.utils), [1] get_params() (ot.utils.BaseEstimator method), [1] get_plan_from_dual() (in module ot.smooth), [1] get_plan_from_semi_dual() (in module ot.smooth), [1] get_random_projections() (in module ot.sliced), [1]

greenkhorn() (in module ot.bregman), [1] gromov_barycenters() (in module ot) (in module ot.gromov) gromov_wasserstein() (in module ot) (in module ot.gromov) gromov_wasserstein2() (in module ot) (in module ot.gromov) gromov_wasserstein_dictionary_learning() (in module ot.gromov) gromov_wasserstein_linear_unmixing() (in module ot.gromov) GW_distance_estimation() (in module ot.gromov) gwggrad() (in module ot.gromov) gwgrad_partial() (in module ot.partial), [1] gwloss() (in module ot.gromov) gwloss_partial() (in module ot.partial), [1]

I

init_matrix() (in module ot.gromov) init_matrix_semirelaxed() (in module ot.gromov) inv() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] inverse_transform() (ot.da.BaseTransport method), [1] (ot.da.LinearTransport method), [1] inverse_transform_labels() (ot.da.BaseTransport method), [1] (ot.da.JCPOTTransport method), [1] is_floating_point() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] isfinite() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

isinf() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] isnan() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] issparse() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

J

JaxBackend (class in ot.backend), [1] jcpot_barycenter() (in module ot.bregman), [1]

JCPOTTransport (class in ot.da), [1] joint_OT_mapping_kernel() (in module ot.da), [1] joint_OT_mapping_linear() (in module ot.da), [1]

K

kernel() (in module ot.utils), [1] kl_div() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

L

label_normalization() (in module ot.utils), [1] laplacian() (in module ot.utils), [1] lazy_plan (ot.utils.OTResult property) lbfgsb_unbalanced() (in module ot.unbalanced), [1] line_search_armijo() (in module ot.optim), [1] LinearGWTransport (class in ot.da), [1] LinearTransport (class in ot.da), [1] linspace() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] list_to_array() (in module ot.utils), [1] log() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

log_ (ot.da.JCPOTTransport attribute), [1] (ot.da.MappingTransport attribute), [1] (ot.da.SinkhornL1l2Transport attribute), [1] (ot.da.SinkhornTransport attribute), [1] (ot.da.UnbalancedSinkhornTransport attribute), [1] logsumexp() (in module ot.dr), [1] (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] loss_dual_entropic() (in module ot.stochastic), [1] loss_dual_quadratic() (in module ot.stochastic), [1]

M

make_1D_gauss() (in module ot.datasets), [1] make_2D_samples_gauss() (in module ot.datasets), [1] make_data_classif() (in module ot.datasets), [1] mapping_ (ot.da.MappingTransport attribute), [1] MappingTransport (class in ot.da), [1] marginal_a (ot.utils.OTResult property) marginal_b (ot.utils.OTResult property) marginals (ot.utils.OTResult property) matmul() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] max() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] max_Omega() (ot.smooth.NegEntropy method), [1] (ot.smooth.Regularization method), [1] (ot.smooth.SparsityConstrained method), [1] (ot.smooth.SquaredL2 method), [1] max_sliced_wasserstein_distance() (in module ot) (in module ot.sliced), [1] maximum() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] mean() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] median() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

meshgrid() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] min() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] minimum() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] mm_unbalanced() (in module ot.unbalanced), [1] mm_unbalanced2() (in module ot.unbalanced), [1] module ot ot.backend ot.bregman ot.coot ot.da ot.datasets ot.dr ot.factored ot.gaussian ot.gnn ot.gromov ot.lp ot.optim ot.partial ot.plot ot.regpath ot.sliced ot.smooth ot.stochastic ot.unbalanced ot.utils ot.weak

N

NegEntropy (class in ot.smooth), [1] norm() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

NumpyBackend (class in ot.backend), [1]

O

Omega() (ot.smooth.NegEntropy method), [1] (ot.smooth.Regularization method), [1] (ot.smooth.SparsityConstrained method), [1] (ot.smooth.SquaredL2 method), [1] ones() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] ot module ot.backend module ot.bregman module ot.coot module ot.da module ot.datasets module ot.dr module ot.factored module ot.gaussian module ot.gnn module ot.gromov module

ot.lp module ot.optim module ot.partial module ot.plot module ot.regpath module ot.sliced module ot.smooth module ot.stochastic module ot.unbalanced module ot.utils module ot.weak module ot_next_gamma() (in module ot.regpath), [1] OTResult (class in ot.utils) outer() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

P

parmap() (in module ot.utils), [1] partial_gromov_wasserstein() (in module ot.partial), [1] partial_gromov_wasserstein2() (in module ot.partial), [1] partial_wasserstein() (in module ot.partial), [1] partial_wasserstein2() (in module ot.partial), [1] partial_wasserstein_lagrange() (in module ot.partial), [1] plan (ot.utils.OTResult property) plan_dual_entropic() (in module ot.stochastic), [1] plan_dual_quadratic() (in module ot.stochastic), [1] plot1D_mat() (in module ot.plot), [1] plot2D_samples_mat() (in module ot.plot), [1] pointwise_gromov_wasserstein() (in module ot.gromov) potential_a (ot.utils.OTResult property) potential_b (ot.utils.OTResult property) potentials (ot.utils.OTResult property) power() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

prod() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] proj_simplex() (in module ot.utils), [1] projC() (in module ot.bregman), [1] projection_robust_wasserstein() (in module ot.dr), [1] projection_simplex() (in module ot.smooth), [1] projection_sparse_simplex() (in module ot.utils), [1] projR() (in module ot.bregman), [1] proportions_ (ot.da.JCPOTTransport attribute), [1]

Q

qr() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

R

rand() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] randn() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] recast_ot_as_lasso() (in module ot.regpath), [1] recast_semi_relaxed_as_lasso() (in module ot.regpath), [1]

register_backend() (in module ot.backend) Regularization (class in ot.smooth), [1] regularization_path() (in module ot.regpath), [1] repeat() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] reshape() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

S

sag_entropic_transport() (in module ot.stochastic), [1] sampled_gromov_wasserstein() (in module ot.gromov) screenkhorn() (in module ot.bregman), [1] searchsorted() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] seed() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] semi_dual_obj_grad() (in module ot.smooth), [1] semi_relaxed_next_gamma() (in module ot.regpath), [1] semi_relaxed_path() (in module ot.regpath), [1] semidiscrete_wasserstein2_unif_circle() (in module ot) (in module ot.lp) semirelaxed_cg() (in module ot.optim), [1] semirelaxed_fused_gromov_wasserstein() (in module ot.gromov) semirelaxed_fused_gromov_wasserstein2() (in module ot.gromov) semirelaxed_gromov_wasserstein() (in module ot.gromov) semirelaxed_gromov_wasserstein2() (in module ot.gromov) set_gradients() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] set_params() (ot.utils.BaseEstimator method), [1] sgd_entropic_regularization() (in module ot.stochastic), [1] sign() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] sinkhorn() (in module ot) (in module ot.bregman), [1] (in module ot.dr), [1] sinkhorn2() (in module ot) (in module ot.bregman), [1] sinkhorn_epsilon_scaling() (in module ot.bregman), [1] sinkhorn_knopp() (in module ot.bregman), [1] sinkhorn_knopp_unbalanced() (in module ot.unbalanced), [1] sinkhorn_l1l2_gl() (in module ot.da), [1] sinkhorn_log() (in module ot.bregman), [1] (in module ot.dr), [1] sinkhorn_lpl1_mm() (in module ot) (in module ot.da), [1] sinkhorn_stabilized() (in module ot.bregman), [1] sinkhorn_stabilized_unbalanced() (in module ot.unbalanced), [1] sinkhorn_unbalanced() (in module ot) (in module ot.unbalanced), [1] sinkhorn_unbalanced2() (in module ot) (in module ot.unbalanced), [1] SinkhornL1l2Transport (class in ot.da), [1] SinkhornLpl1Transport (class in ot.da), [1] SinkhornTransport (class in ot.da), [1] sliced_wasserstein_distance() (in module ot) (in module ot.sliced), [1] sliced_wasserstein_sphere() (in module ot) (in module ot.sliced), [1] sliced_wasserstein_sphere_unif() (in module ot) (in module ot.sliced), [1] smooth_ot_dual() (in module ot.smooth), [1]

smooth_ot_semi_dual() (in module ot.smooth), [1] solve() (in module ot) (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] solve_1d_linesearch_quad() (in module ot.optim), [1] solve_dual() (in module ot.smooth), [1] solve_dual_entropic() (in module ot.stochastic), [1] solve_gromov_linesearch() (in module ot.gromov) solve_semi_dual() (in module ot.smooth), [1] solve_semi_dual_entropic() (in module ot.stochastic), [1] solve_semirelaxed_gromov_linesearch() (in module ot.gromov) sort() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] sort2() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] sparse_plan (ot.utils.OTResult property) SparsityConstrained (class in ot.smooth), [1] split_classes() (in module ot.dr), [1] sqrt() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] sqrtm() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] SquaredL2 (class in ot.smooth), [1] squeeze() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] stack() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] status (ot.utils.OTResult property) std() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] sum() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

T

take_along_axis() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] tensor_product() (in module ot.gromov) TensorflowBackend (class in ot.backend), [1] TFGWPooling (class in ot.gnn) tic() (in module ot) (in module ot.utils), [1] tile() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] to_numpy() (in module ot.backend), [1] (ot.backend.Backend method), [1] toc() (in module ot) (in module ot.utils), [1] tocsr() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

todense() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] toq() (in module ot) (in module ot.utils), [1] TorchBackend (class in ot.backend), [1] trace() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] transform() (ot.da.BaseTransport method), [1] (ot.da.JCPOTTransport method), [1] (ot.da.LinearTransport method), [1] (ot.da.MappingTransport method), [1] transform_labels() (ot.da.BaseTransport method), [1] (ot.da.JCPOTTransport method), [1] transpose() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1] TWPooling (class in ot.gnn)

U

UnbalancedSinkhornTransport (class in ot.da), [1] UndefinedParameter unif() (in module ot) (in module ot.utils), [1] unique() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

unmix() (in module ot.bregman), [1] update_feature_matrix() (in module ot.gromov) update_kl_loss() (in module ot.gromov) update_square_loss() (in module ot.gromov)

V

value (ot.utils.OTResult property)

value_linear (ot.utils.OTResult property)

W

wasserstein_1d() (in module ot) (in module ot.lp) wasserstein_circle() (in module ot) (in module ot.lp) wasserstein_distance_to_templates() (in module ot.gnn) wda() (in module ot.dr), [1] weak_optimal_transport() (in module ot) (in module ot.weak), [1]

where() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

Z

zero_pad() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]

zeros() (ot.backend.Backend method), [1] (ot.backend.CupyBackend method), [1] (ot.backend.JaxBackend method), [1] (ot.backend.NumpyBackend method), [1] (ot.backend.TensorflowBackend method), [1] (ot.backend.TorchBackend method), [1]