tf.tpu.experimental.embedding.FeatureConfig

Данные конфигурации для одного признака вложения.

tf.tpu.experimental.embedding.FeatureConfig(
    table: tf.tpu.experimental.embedding.TableConfig,
    max_sequence_length: int = 0,
    validate_weights_and_indices: bool = True,
    output_shape: Optional[Union[List[int], tf.TensorShape]] = None,
    name: Optional[Text] = None
)

Этот класс хранит данные конфигурации для одного признака вложения. Основное применение — назначение признаков 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 таблицы и три признака. Первые два признака будут искать в первой таблице, а третий — во второй таблице.

Также можно указать форму вывода для каждого признака. Форма вывода должна соответствовать ожидаемой форме активации, исключая размер таблицы. Для плотных и разреженных тензоров форма вывода должна совпадать с формой входных данных, за исключением последнего измерения. Для разреженных тензоров форма вывода может отличаться от формы входных данных.

Примечание: Параметр max_sequence_length будет использоваться только в том случае, если входной тензор имеет ранг 2, а output_shape не задан в конфигурации признака.

При подаче признаков в embedding.enqueue они могут быть tf.Tensor, tf.SparseTensor или tf.RaggedTensor. Когда аргумент max_sequence_length равен 0 (значение по умолчанию), ожидается вывод embedding.dequeue для данного признака с формой (batch_size, dim). Если max_sequence_length больше 0, признак вкладывается как последовательность и дополняется до заданной длины. Форма вывода для этого признака будет (batch_size, max_sequence_length, dim).