tf.lookup.experimental.DatasetInitializer
Создаёт инициализатор таблицы из tf.data.Dataset.
tf.lookup.experimental.DatasetInitializer(
dataset
)
Пример использования:
keys = tf.data.Dataset.range(100)
values = tf.data.Dataset.range(100).map(
lambda x: string_ops.as_string(x * 2))
ds = tf.data.Dataset.zip((keys, values))
init = tf.lookup.experimental.DatasetInitializer(ds)
table = tf.lookup.StaticHashTable(init, "")
output = table.lookup([0, 1, 2])
assertEquals(outputs, ["0", "2", "4"])
Возбуждает исключение ValueError, если dataset не соответствует спецификациям.
| Аргументы | |
|---|---|
dataset | Объект tf.data.Dataset, который генерирует кортежи скаляров. Первый скаляр рассматривается как ключ, а второй — как значение. |
| Атрибуты | |
|---|---|
dataset | Объект tf.data.Dataset, который генерирует кортежи скаляров. Первый скаляр рассматривается как ключ, а второй — как значение. |
key_dtype | Ожидаемый тип данных ключа таблицы. |
value_dtype | Ожидаемый тип данных значения таблицы. |
Методы
initialize
initialize(
table
)
Возвращает оператор инициализации таблицы.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/lookup/experimental/DatasetInitializer