Should the variable A be taken a square root?

[fazhmshily]

In equation 10 of the paper, A_t is taken to the be square root of square(S_re)+square(S_im). However, in line 158 of train/itosfm.py, A_t is taken to be square(S_re)+square(S_im), without the square root. Is this a typo or my understanding is wrong? Thank you very much!