使用tf.compat.v1转换TF1代码时出现contrib模块不存在的属性错误

错误原因 #

TensorFlow 2.x 彻底移除了 contrib 模块——哪怕是 tf.compat.v1 兼容层也不再包含该模块。tf.contrib 是TF1时代的实验性功能集合，TF2发布时将其中成熟功能整合到了核心API，其余未成熟的则直接弃用，因此你无法通过 tf.compat.v1.contrib 访问任何内容。

解决方法 #

你代码里用到的 xavier_initializer 其实就是Glorot均匀初始化，TF2的核心API已经提供了官方替代方案，直接替换即可：

替换方案1（原生TF2 API） #

W = tf.Variable(
    tf.keras.initializers.GlorotUniform()(shape=[num_filters_total, num_classes]),
    name="W"
)

替换方案2（兼容TF1风格） #

如果想保留TF1的get_variable写法，可改用TF2兼容层中的初始化器：

W = tf.compat.v1.get_variable(
    "W",
    shape=[num_filters_total, num_classes],
    initializer=tf.compat.v1.keras.initializers.GlorotUniform()
)

两种方案效果完全一致，都是实现Xavier初始化的权重生成逻辑。

内容的提问来源于stack exchange，提问作者kiriloff