Calculate Batch Pairwise Sinkhorn Distance using geomloss in PyTorch

Question

I have two tensors and both are of same shape. I want to calculate pairwise sinkhorn distance using geomloss.

What i have tried:

impor