Hi, in `line 43` of `mmd_loss.py`, you wrote `dot(x, x.T)`, I think it should be `dot(x.T, x)`, correct?