Python tf.compat.v1.tables_initializer用法及代碼示例

返回一個初始化默認圖的所有表的操作。

用法

tf.compat.v1.tables_initializer(
    name='init_all_tables'
)

遷移到 TF2

警告：這個 API 是為 TensorFlow v1 設計的。繼續閱讀有關如何從該 API 遷移到本機 TensorFlow v2 等效項的詳細信息。見TensorFlow v1 到 TensorFlow v2 遷移指南有關如何遷移其餘代碼的說明。

即刻執行和 tf.function 不再需要 tf.compat.v1.tables_initializer 。在 TF2 中，當創建像 tf.lookup.StaticHashTable 這樣的可初始化表時，該表將在創建時自動初始化。

使用示例之前和之後

前：

with tf.compat.v1.Session():
  init = tf.compat.v1.lookup.KeyValueTensorInitializer(['a', 'b'], [1, 2])
  table = tf.compat.v1.lookup.StaticHashTable(init, default_value=-1)
  tf.compat.v1.tables_initializer().run()
  result = table.lookup(tf.constant(['a', 'c'])).eval()
result
array([ 1, -1], dtype=int32)

後：

init = tf.lookup.KeyValueTensorInitializer(['a', 'b'], [1, 2])
table = tf.lookup.StaticHashTable(init, default_value=-1)
table.lookup(tf.constant(['a', 'c'])).numpy()
array([ 1, -1], dtype=int32)