layers.py 文件源码

def reset_parameters(self):

        linear_gain = weight_init.calculate_gain('linear')

        weight_init.xavier_uniform(self.W_x.data, linear_gain)
        weight_init.xavier_uniform(self.W_y.data, linear_gain)
        weight_init.xavier_uniform(self.W_z.data, linear_gain)