urnn.py 文件源码

def reset_parameters(self):
        """
        Initialize parameters
        """
        init.uniform(self.thetaA, a=-0.1, b=0.1)
        init.uniform(self.thetaB, a=-0.1, b=0.1)
        init.uniform(self.U, a=-0.1, b=0.1)
        init.constant(self.bias.data, val=0)