Модуль: tf.tpu.experimental.embedding

Публичный API для пространства имен tf._api.v2.tpu.experimental.embedding

Классы

class Adagrad: Параметры оптимизации для Adagrad с TPU вложениями.

class AdagradMomentum: Параметры оптимизации для Adagrad + Momentum с TPU вложениями.

class Adam: Параметры оптимизации для Adam с TPU вложениями.

class FTRL: Параметры оптимизации для FTRL с TPU вложениями.

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

class QuantizationConfig: Настройки для моделирования квантования таблицы вложений tpu.

class RowIdInitializer: Инициализатор, который инициализирует таблицу идентификаторами словаря.

class SGD: Параметры оптимизации для стохастического градиентного спуска для TPU вложений.

class TPUEmbedding: API среднего уровня TPUEmbedding.

class TPUEmbeddingForServing: API среднего уровня TPUEmbedding, работающий на CPU для предоставления услуг.

class TPUEmbeddingV0: API среднего уровня TPUEmbedding, работающий на TPU без ускорителя вложений.

class TPUEmbeddingV2: API среднего уровня TPUEmbedding, работающий на TPU с ускорителем разреженных ядер.

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

Функции

serving_embedding_lookup(...): Применение стандартных операций поиска с конфигурациями tf.tpu.experimental.embedding.