I am a newbie in pytorch. It seems that the notebook use the self.Q and self.R for all rounds while the equation use Q_i and R_i for i th round?