Операция, позволяющая дифференцировать встраивания TPU.
Эта операция просто возвращает свои первые входные данные, которые, как предполагается, были вырезаны из тензоров, возвращенных TPUEmbeddingDequeueActivations. Наличие этой операции и ее первого аргумента, являющегося обучаемой переменной, позволяет автоматически дифференцировать графы, содержащие внедрения, с помощью библиотек Python для встраивания TPU.
Константы
| Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
| Вывод <TFloat32> | какВывод () Возвращает символический дескриптор тензора. |
| статические EmbeddingActivations | create ( Область действия, Операнд <TFloat32> embeddingVariable, Операнд <TFloat32> слайсактивация, Длинный tableId, Длинный идентификатор поиска) Фабричный метод для создания класса, обертывающего новую операцию EmbeddingActivations. |
| Вывод <TFloat32> | выход () |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <TFloat32> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static EmbeddingActivations create (область действия , операнд < TFloat32 > embeddingVariable, операнд < TFloat32 > слайсактивации, длинный tableId, длинный поиск)
Фабричный метод для создания класса, обертывающего новую операцию EmbeddingActivations.
Параметры
| объем | текущий объем |
|---|---|
| встраиваниеПеременная | Обучаемая переменная, позволяющая оптимизаторам находить эту операцию. |
| нарезанные активации | Встраивание активирует Tensor для возврата. |
| идентификатор таблицы | Идентификатор таблицы в конфигурации уровня внедрения, на основе которой были вычислены эти активации. |
| идентификатор поиска | Идентификатор набора индексов внедрения, вызвавших эти активации. |
Возврат
- новый экземпляр EmbeddingActivations