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

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

Классы

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

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

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

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

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

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

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

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

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

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

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

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

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

Функции

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