tf.tpu.experimental.embedding.TableConfig

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

tf.tpu.experimental.embedding.TableConfig(
    vocabulary_size: int,
    dim: int,
    initializer: Optional[Callable[[Any], None]] = None,
    optimizer: Optional[_Optimizer] = None,
    combiner: Text = 'mean',
    name: Optional[Text] = 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 таблицы и три признака. Первые два признака будут искаться в первой таблице, а третий – во второй таблице.