将 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

TF2 的替代品tf.contrib.layers.xavier_initializer()是 tf.keras.initializers.glorot_normal（Xavier 和 Glorot 是同一初始化算法的 2 个名称，指的是一位名为 Xavier Glorot 的研究人员）文档链接。

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