tf.compat.v1.contrib.framework.assign_from_values()改为tensorflow2.x兼容语句

在 TensorFlow 2.x 中，`tf.compat.v1.contrib.framework.assign_from_values()` 方法已经被移除，可以使用 `tf.train.load_checkpoint()` 方法来加载权重文件，并使用 `tf.Variable.assign()` 方法将权重值赋给变量。以下是 TensorFlow 2.x 兼容的代码：

import tensorflow as tf

checkpoint_path = "path/to/checkpoint"
var_dict = tf.train.load_checkpoint(checkpoint_path).get_variable_to_shape_map()

for var_name, shape in var_dict.items():
    var = tf.Variable(tf.zeros(shape), name=var_name)  # 创建变量
    value = tf.train.load_variable(checkpoint_path, var_name)  # 加载权重值
    var.assign(value)  # 赋值

其中，`checkpoint_path` 表示权重文件的路径，`var_dict` 是一个字典，保存了权重文件中所有变量的名称和形状。在循环中，我们首先使用 `tf.Variable()` 方法创建变量，然后使用 `tf.train.load_variable()` 方法加载权重值，最后使用 `var.assign()` 方法将权重值赋给变量。