RecvTPUEmbeddingActivations

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

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

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

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

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

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

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

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

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

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

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

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

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