TPUEmbeddingActivations

публичный финальный класс TPUEmbeddingActivations

Операция, позволяющая дифференцировать встраивания TPU.

Эта операция просто возвращает свои первые входные данные, которые, как предполагается, были вырезаны из тензоров, возвращенных TPUEmbeddingDequeueActivations. Наличие этой операции и ее первого аргумента, являющегося обучаемой переменной, позволяет автоматически дифференцировать графы, содержащие внедрения, с помощью библиотек Python для встраивания TPU.

Публичные методы

Выход <С плавающей запятой>
какВывод ()
Возвращает символический дескриптор тензора.
статические TPUEmbeddingActivations
create ( Область действия, Операнд <Float> embeddingVariable, Операнд <Float> слайсактивация, Длинный tableId, Длинный идентификатор поиска)
Фабричный метод для создания класса, обертывающего новую операцию TPUEmbeddingActivations.
Выход <С плавающей запятой>

Унаследованные методы

Публичные методы

общедоступный вывод <Float> asOutput ()

Возвращает символический дескриптор тензора.

Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.

public static TPUEmbeddingActivations create (область действия , операнд <Float> embeddingVariable, операнд <Float> слайсактивация, длинный tableId, длинный поискId)

Фабричный метод для создания класса, обертывающего новую операцию TPUEmbeddingActivations.

Параметры
объем текущий объем
встраиваниеПеременная Обучаемая переменная, позволяющая оптимизаторам находить эту операцию.
нарезанные активации Встраивание активирует Tensor для возврата.
идентификатор таблицы Идентификатор таблицы в конфигурации уровня внедрения, на основе которой были вычислены эти активации.
идентификатор поиска Идентификатор набора индексов внедрения, вызвавших эти активации.
Возврат
  • новый экземпляр TPUEmbeddingActivations

публичный вывод <Float> вывод ()