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, "")
table.lookup(tf.constant([0, 1, 2], dtype=tf.int64)).numpy()
array([b'0', b'2', b'4'], dtype=object)
Возникает исключение 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.4/api_docs/python/tf/lookup/experimental/DatasetInitializer