Hi,
I am using the Geomloss package to compute the Sinkhorn  distance. my inputs are two distributionS(DATA1 AND DATA2) with batch size of 64 and 99 dimension. how I can pass them into the loss to compute , should I pass them as a vector with size of 649*9?
    loss = SamplesLoss(loss="sinkhorn", p=2, blur=.5)
    L = loss(DATA1.float(),DATA2.float())