Index

A

a_to_b (ot.utils.OTResult property), [1] 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), [1] Backend (class in ot.backend), [1] BAPG_fused_gromov_wasserstein() (in module ot.gromov) BAPG_fused_gromov_wasserstein2() (in module ot.gromov) BAPG_gromov_wasserstein() (in module ot.gromov) BAPG_gromov_wasserstein2() (in module ot.gromov) barycenter() (in module ot) (in module ot.bregman) (in module ot.lp) barycenter_debiased() (in module ot.bregman) barycenter_sinkhorn() (in module ot.bregman) barycenter_stabilized() (in module ot.bregman) barycenter_unbalanced() (in module ot) (in module ot.unbalanced) barycenter_unbalanced_sinkhorn() (in module ot.unbalanced)

barycenter_unbalanced_stabilized() (in module ot.unbalanced) 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_barycenter() (in module ot.gaussian), [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), [1] 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_lr_sqeuclidean_matrix() (in module ot.lowrank), [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) convolutional_barycenter2d_debiased() (in module ot.bregman)

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] det() (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] detach() (ot.backend.Backend 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] dist_bures_squared() (in module ot.gmm), [1] distribution_estimation_uniform() (in module ot.da), [1] div_between_product() (in module ot.gromov) div_to_product() (in module ot.gromov) 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_1d_sorted() (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_barycenter() (in module ot.gaussian), [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) empirical_sinkhorn2() (in module ot.bregman) empirical_sinkhorn2_geomloss() (in module ot.bregman) empirical_sinkhorn_divergence() (in module ot.bregman) 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.gromov) (in module ot.partial), [1] entropic_partial_gromov_wasserstein2() (in module ot.gromov) (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.NearestBrenierPotential 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]

format_partitioned_graph() (in module ot.gromov) format_partitioned_samples() (in module ot.gromov) 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) 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) fused_unbalanced_across_spaces_cost() (in module ot.gromov) fused_unbalanced_across_spaces_divergence() (in module ot.gromov) fused_unbalanced_gromov_wasserstein() (in module ot.gromov) fused_unbalanced_gromov_wasserstein2() (in module ot.gromov)

G

gaussian_gromov_wasserstein_distance() (in module ot.gaussian), [1] gaussian_gromov_wasserstein_mapping() (in module ot.gaussian), [1] gaussian_logpdf() (in module ot.gmm), [1] gaussian_pdf() (in module ot.gmm), [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) geometricMean() (in module ot.bregman) get_available_backend_implementations() (in module ot.backend), [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_graph_partition() (in module ot.gromov) get_graph_representants() (in module ot.gromov) get_lowrank_lazytensor() (in module ot.utils), [1] get_parameter_pair() (in module ot.utils), [1] get_params() (ot.utils.BaseEstimator method), [1] get_partition_and_representants_samples() (in module ot.gromov) 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] gmm_ot_apply_map() (in module ot.gmm), [1] gmm_ot_loss() (in module ot.gmm), [1] gmm_ot_plan() (in module ot.gmm), [1] gmm_ot_plan_density() (in module ot.gmm), [1] gmm_pdf() (in module ot.gmm), [1] greenkhorn() (in module ot.bregman) 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] (ot.da.SinkhornTransport method), [1] inverse_transform_labels() (ot.da.BaseTransport method), [1] (ot.da.JCPOTTransport method), [1] is_all_finite() (in module ot.utils), [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)

JCPOTTransport (class in ot.da), [1] joint_OT_mapping_kernel() (in module ot.mapping), [1] joint_OT_mapping_linear() (in module ot.mapping), [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] labels_to_masks() (in module ot.utils), [1] laplacian() (in module ot.utils), [1] lazy_plan (ot.utils.OTResult attribute), [1] (ot.utils.OTResult property), [1] LazyTensor (class in ot.utils), [1] lbfgsb_unbalanced() (in module ot.unbalanced) lbfgsb_unbalanced2() (in module ot.unbalanced) 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.utils.OTResult property), [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] lowrank_gromov_wasserstein_samples() (in module ot) (in module ot.gromov) lowrank_sinkhorn() (in module ot) (in module ot.lowrank), [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 attribute), [1] (ot.utils.OTResult property), [1] marginal_b (ot.utils.OTResult attribute), [1] (ot.utils.OTResult property), [1] marginals (ot.utils.OTResult attribute), [1] (ot.utils.OTResult property), [1] 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) mm_unbalanced2() (in module ot.unbalanced) module ot ot.backend ot.bregman ot.coot ot.da ot.datasets ot.dr ot.factored ot.gaussian ot.gmm ot.gnn ot.gromov ot.lowrank ot.lp ot.mapping ot.optim ot.partial ot.plot ot.regpath ot.sliced ot.smooth ot.stochastic ot.unbalanced ot.utils ot.weak

N

nan_to_num() (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] nearest_brenier_potential_fit() (in module ot.mapping), [1] nearest_brenier_potential_predict_bounds() (in module ot.mapping), [1]

NearestBrenierPotential (class in ot.da), [1] 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.gmm module ot.gnn module ot.gromov module

ot.lowrank module ot.lp module ot.mapping 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), [1] 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_cg() (in module ot.optim), [1] partial_fused_gromov_wasserstein() (in module ot.gromov) partial_fused_gromov_wasserstein2() (in module ot.gromov) partial_gromov_wasserstein() (in module ot.gromov) (in module ot.partial), [1] partial_gromov_wasserstein2() (in module ot.gromov) (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 attribute), [1] (ot.utils.OTResult property), [1] 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 attribute), [1] (ot.utils.OTResult property), [1] potential_b (ot.utils.OTResult attribute), [1] (ot.utils.OTResult property), [1]

potentials (ot.utils.OTResult attribute), [1] (ot.utils.OTResult property), [1] 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_SDP() (in module ot.utils), [1] proj_simplex() (in module ot.utils), [1] projC() (in module ot.bregman) 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) 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]

quantized_fused_gromov_wasserstein() (in module ot.gromov) quantized_fused_gromov_wasserstein_partitioned() (in module ot.gromov) quantized_fused_gromov_wasserstein_samples() (in module ot.gromov)

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] reduce_lazytensor() (in module ot.utils), [1]

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] rescale_for_imshow_plot() (in module ot.plot), [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) 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_fgw_barycenters() (in module ot.gromov) semirelaxed_fused_gromov_wasserstein() (in module ot.gromov) semirelaxed_fused_gromov_wasserstein2() (in module ot.gromov) semirelaxed_gromov_barycenters() (in module ot.gromov) semirelaxed_gromov_wasserstein() (in module ot.gromov) semirelaxed_gromov_wasserstein2() (in module ot.gromov) semirelaxed_init_plan() (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) (in module ot.dr), [1] sinkhorn2() (in module ot) (in module ot.bregman) sinkhorn_epsilon_scaling() (in module ot.bregman) sinkhorn_knopp() (in module ot.bregman) sinkhorn_knopp_unbalanced() (in module ot.unbalanced) sinkhorn_l1l2_gl() (in module ot.da), [1] sinkhorn_log() (in module ot.bregman) (in module ot.dr), [1] sinkhorn_lpl1_mm() (in module ot) (in module ot.da), [1] sinkhorn_stabilized() (in module ot.bregman) sinkhorn_stabilized_unbalanced() (in module ot.unbalanced) sinkhorn_unbalanced() (in module ot) (in module ot.unbalanced) sinkhorn_unbalanced2() (in module ot) (in module ot.unbalanced) sinkhorn_unbalanced_translation_invariant() (in module ot.unbalanced) 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] slogdet() (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]

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() (in module ot) solve_gromov_linesearch() (in module ot.gromov) solve_partial_gromov_linesearch() (in module ot.gromov) solve_sample() (in module ot) 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 attribute), [1] (ot.utils.OTResult property), [1] 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), [1] 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] (ot.da.NearestBrenierPotential method), [1] (ot.da.SinkhornTransport 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

unbalanced_co_optimal_transport() (in module ot.gromov) unbalanced_co_optimal_transport2() (in module ot.gromov) 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) uot_cost_matrix() (in module ot.gromov) uot_parameters_and_measures() (in module ot.gromov) update_barycenter_feature() (in module ot.gromov) update_barycenter_structure() (in module ot.gromov)

V

value (ot.utils.OTResult attribute), [1] (ot.utils.OTResult property), [1] value_linear (ot.utils.OTResult attribute), [1] (ot.utils.OTResult property), [1]

value_quad (ot.utils.OTResult attribute), [1] (ot.utils.OTResult property), [1]

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]