tf.tpu.experimental.embedding.TableConfig

Данные конфигурации для одной таблицы встраивания.

tf.tpu.experimental.embedding.TableConfig(
    vocabulary_size, dim, initializer, optimizer=None, combiner='mean', name=None
)

Этот класс содержит данные конфигурации для одной таблицы встраивания. Он используется в качестве параметра table для tf.tpu.experimental.embedding.FeatureConfig. Несколько объектов tf.tpu.experimental.embedding.FeatureConfig могут использовать один и тот же объект tf.tpu.experimental.embedding.TableConfig. В этом случае для этих поисков по признакам будет создана общая таблица.

table_config_one = tf.tpu.experimental.embedding.TableConfig(
    vocabulary_size=...,
    dim=...)
table_config_two = tf.tpu.experimental.embedding.TableConfig(
    vocabulary_size=...,
    dim=...)
feature_config = {
    'feature_one': tf.tpu.experimental.embedding.FeatureConfig(
        table=table_config_one),
    'feature_two': tf.tpu.experimental.embedding.FeatureConfig(
        table=table_config_one),
    'feature_three': tf.tpu.experimental.embedding.FeatureConfig(
        table=table_config_two)}
embedding = tf.tpu.experimental.embedding.TPUEmbedding(
    feature_config=feature_config,
    batch_size=...
    optimizer=tf.tpu.experimental.embedding.Adam(0.1))

В приведенной выше конфигурации 2 таблицы и три признака. Первые два признака будут найдены в первой таблице, а третий — во второй.