Index

A

adjoint_sinkhorn_vjp() (in module torchgw._sinkhorn)

B

build_knn_graph() (in module torchgw)

G

gw_cost_estimate() (in module torchgw._loop)

gw_loop() (in module torchgw._loop)

J

joint_embedding() (in module torchgw)

S

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

sampled_lowrank_gw() (in module torchgw) sinkhorn() (in module torchgw._sinkhorn) sinkhorn_differentiable() (in module torchgw._sinkhorn)