InfeedEnqueueTuple

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

Подает несколько значений Tensor в вычисления в виде кортежа XLA.

Вложенные классы

сорт InfeedEnqueueTuple.Options Необязательные атрибуты для InfeedEnqueueTuple

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

статический InfeedEnqueueTuple
create (область области действия , входные данные Iterable< Operand <?>>, фигуры List< Shape >, параметры ... )
Фабричный метод для создания класса, обертывающего новую операцию InfeedEnqueueTuple.
статический InfeedEnqueueTuple.Options
deviceOrdinal (длинный номер устройства)
статический InfeedEnqueueTuple.Options
макеты (макеты List<Long>)

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

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

public static InfeedEnqueueTuple create (область области действия , входные данные Iterable< Operand <?>>, фигуры List< Shape >, параметры... параметры)

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

Параметры
объем текущий объем
входы Список тензоров, которые будут предоставлены с помощью механизма подачи.
формы Формы каждого тензора во «входах».
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр InfeedEnqueueTuple

общедоступный статический InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)

Параметры
устройствоПорядковый номер Используемое устройство ТПУ. Это значение должно быть -1, когда Op работает на устройстве TPU, и >= 0, когда Op работает на устройстве CPU.

общедоступные статические макеты InfeedEnqueueTuple.Options (макеты List<Long>)

Параметры
макеты Вектор, содержащий запрошенный макет в последовательности от младшего к старшему для всех фигур кортежа в том порядке, в котором фигуры появляются во входных данных «фигуры». Элементам макета для подформы можно установить значение -1, и в этом случае соответствующий макет будет вычислен с помощью операции подачи.