Index

_

_sinkhorn_differentiable() (in module torchgw._solver)

_sinkhorn_torch() (in module torchgw._solver)

B

build_knn_graph() (in module torchgw)

G

get_device() (in module torchgw._utils)

J

joint_embedding() (in module torchgw)

S

sample_pairs_from_plan() (in module torchgw._sampling) sample_pairs_gpu() (in module torchgw._sampling)

sampled_gw() (in module torchgw) sampled_lowrank_gw() (in module torchgw)