Sinkhorn Iteration

[zhenyuzhouiu]

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!