RecvTPUEmbeddingActivations

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

Операция, которая получает активации встраивания в TPU.

Система TPU выполняет поиск по внедрению и агрегацию, заданную аргументами TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch. Результаты этих агрегаций видны в Tensorflow Graph как выходные данные операции RecvTPUEmbeddingActivations. Эта операция возвращает список, содержащий один тензор активаций для каждой таблицы, указанной в модели. В графе TPU может быть не более одной операции RecvTPUEmbeddingActivations.

Константы

Нить OP_NAME Название этой операции, известное основному движку TensorFlow.

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

статические RecvTPUEmbeddingActivations
create (область области действия , длинное количество выходов, строковая конфигурация)
Фабричный метод для создания класса, обертывающего новую операцию RecvTPUEmbeddingActivations.
Итератор< Операнд <TFloat32> >
Список < Вывод < TFloat32 >>
выходы ()
TensorList активаций внедрения, содержащий по одному тензору на каждую таблицу внедрения в модели.

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

Константы

общедоступная статическая финальная строка OP_NAME

Название этой операции, известное основному движку TensorFlow.

Постоянное значение: «RecvTPUEmbeddingActivations».

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

public static RecvTPUEmbeddingActivations create (область области , Long numOutputs, строковая конфигурация)

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

Параметры
объем текущий объем
числовыходов Количество выходных тензоров активации, равное количеству таблиц встраивания в модель.
конфигурация Сериализованный прототип TPUEmbeddingConfiguration.
Возврат
  • новый экземпляр RecvTPUEmbeddingActivations

общедоступный итератор < операнд < TFloat32 >> итератор ()

общедоступный список< Вывод <TFloat32> > выходы ()

TensorList активаций внедрения, содержащий по одному тензору на каждую таблицу внедрения в модели.