get_initializer

功能描述

获取初始化iterator的TensorFlow算子（tf.Operation），该算子需要通过sess.run()使用。

函数原型

def get_initializer(is_training)

参数说明

返回值说明

• 成功：返回初始化iterator的TensorFlow算子（tf.Operation）。
• 失败：抛出异常。

使用示例

from mx_rec.util.initialize import get_initializer

with tf.compat.v1.Session(config=sess_config()) as sess:
    sess.run(get_initializer(True))

参考资源

接口调用流程及示例，参见自动改图。