Spec-Zone.ru › TensorFlow 1.15
tf.lookup.StaticHashTable Обобщённая хеш-таблица, которая становится неизменяемой после инициализации.
Наследуется от: StaticHashTable
tf.lookup.StaticHashTable(
initializer, default_value, name=None
)
При выполнении в режиме графа, вы должны оценить тензор, возвращаемый tf.tables_initializer() , прежде чем оценивать тензор, возвращаемый методом этого класса lookup(). Пример использования в режиме графа:
keys_tensor = tf.constant([1, 2])
vals_tensor = tf.constant([3, 4])
input_tensor = tf.constant([1, 5])
table = tf.lookup.StaticHashTable(
tf.lookup.KeyValueTensorInitializer(keys_tensor, vals_tensor), -1)
out = table.lookup(input_tensor)
with tf.Session() as sess:
sess.run(tf.tables_initializer())
print(sess.run(out))
В режиме eager, нет необходимости в специальном коде для инициализации таблицы. Пример использования в режиме eager:
tf.enable_eager_execution()
keys_tensor = tf.constant([1, 2])
vals_tensor = tf.constant([3, 4])
input_tensor = tf.constant([1, 5])
table = tf.lookup.StaticHashTable(
tf.lookup.KeyValueTensorInitializer(keys_tensor, vals_tensor), -1)
print(table.lookup(input_tensor))
Аргументы
initializer Инициализатор таблицы для использования. См. HashTable ядро для поддерживаемых типов ключей и значений.
default_value Значение, используемое, если ключ отсутствует в таблице.
name Имя операции (необязательно).
Атрибуты
default_value Значение по умолчанию таблицы.
initializer
key_dtype Тип данных ключа таблицы.
name Имя таблицы.
resource_handle Возвращает дескриптор ресурса, связанный с этим ресурсом.
value_dtype Тип данных значений таблицы.
Методы export Просмотреть исходный код
export(
name=None
)
Возвращает тензоры всех ключей и значений в таблице.
Аргументы
name Имя операции (необязательно).
Возвращаемое значение Пара тензоров, первый тензор содержит все ключи, а второй тензор содержит все значения в таблице.
lookup Просмотреть исходный код
lookup(
keys, name=None
)
Ищет keys в таблице, возвращает соответствующие значения.
default_value используется для ключей, отсутствующих в таблице.
Аргументы
keys Ключи для поиска. Может быть либо SparseTensor, либо плотным Tensor.
name Имя операции (необязательно).
Возвращаемое значение SparseTensor если ключи являются разреженными, в противном случае плотный Tensor.
Исключения
TypeError когда keys или default_value не соответствуют типам данных таблицы.
size Просмотреть исходный код
size(
name=None
)
Вычисляет количество элементов в этой таблице.
Аргументы
name Имя операции (необязательно).
Возвращаемое значение Скалярный тензор, содержащий количество элементов в этой таблице.