Module: tf.compat.v1.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 без ускорителя Embedding.

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

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

Функции

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