将tf.contrib.layers.xavier_初始值设定项（）更改为2.0.0

Question

我怎样才能改变

tf.contrib.layers.xavier_initializer()

至tf版本>；=2.0.0 ??

所有代码：

W1 = tf.get_variable("W1", shape=[self.input_size, h_size],
                             initializer=tf.contrib.layers.xavier_initializer())

Answer 1

TF2对tf.contrib.layers.xavier_initializer()的替换为tf.keras.initializers.glorot_normal（Xavier和Glorot是同一初始值设定项算法的两个名称）documentation link

如果出于某些兼容性原因，数据类型很重要-请使用tf.compat.v1.keras.initializers.glorot_normal