Thank you for your work! It gives me a more clear understanding of OT.
But I am new to the Sinkhorn iteration algorithm, and as I understand it, the iteration formula is as follows:

while K is:

However, on your code, I notice the formula is:

I have tried to derive the formula, but I cannot get the same equation.
I hope you can give me an answer, thanks!
Thank you for your work! It gives me a more clear understanding of OT.
But I am new to the Sinkhorn iteration algorithm, and as I understand it, the iteration formula is as follows:


while K is:
However, on your code, I notice the formula is:

I have tried to derive the formula, but I cannot get the same equation.
I hope you can give me an answer, thanks!