将 tf.contrib.layers.xavier_initializer() 更改为 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

tf.contrib.layers.xavier_initializer() 的 TF2 替代品是 tf.keras.initializers.glorot_normal（Xavier 和 Glorot 是同一初始化算法的两个名称）documentation link。

如果 dtype 出于某些兼容性原因很重要 - 使用 tf.compat.v1.keras.initializers.glorot_normal

change tf.contrib.layers.xavier_initializer() to 2.0.0