tensorflow - tf.contrib.lookup.index_table_from_tensor 的选项

Question

我听说 TF 2.0 将删除 tf.contrib，TF 1.12 中 tf.contrib.lookup.index_table_from_tensor 函数的最佳替代方案是什么？

Answer 1

tf.contrib.lookup 中的许多功能已迁移到 tf.lookup ，虽然我不相信 tf.contrib.lookup.index_table_from_tensor 有 1:1 的映射。迄今为止，可用的端点是:

• KeyValueTensorInitializer - 给定 keys 和 values 张量的表初始化器。
• StaticHashTable - 通用哈希表实现。
• StaticVocabularyTable - 字符串到 ID 表包装器，用于将词汇外键分配给存储桶。
• TextFileInitializer - 来自文本文件的表初始化程序。
• experimental.DenseHashTable - 使用张量作为后备存储的通用可变哈希表实现。

您可以在 Sunsetting tf.contrib RFC 中找到有关 tf.contrib 弃用和功能迁移的更多信息.