Модуль: tf.tpu.experimental.embedding
Общедоступный API для пространства имён tf.tpu.experimental.embedding.
Классы
class Adagrad: Параметры оптимизации для Adagrad с встраиванием TPU.
class AdagradMomentum: Параметры оптимизации для Adagrad + Momentum с встраиванием TPU.
class Adam: Параметры оптимизации для Adam с встраиванием TPU.
class FTRL: Параметры оптимизации для FTRL с встраиванием TPU.
class FeatureConfig: Данные конфигурации для одного встраиваемого признака.
class SGD: Параметры оптимизации для стохастического градиентного спуска для встраивания TPU.
class TPUEmbedding: API среднего уровня TPUEmbedding.
class TPUEmbeddingForServing: API среднего уровня TPUEmbedding, работающий на CPU для обработки запросов.
class TPUEmbeddingV0: API среднего уровня TPUEmbedding, работающий на TPU без ускорителя встраивания.
class TableConfig: Данные конфигурации для одной таблицы встраивания.
Функции
serving_embedding_lookup(...): Применение стандартных операций поиска с конфигурациями tf.tpu.experimental.embedding.
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/tpu/experimental/embedding