Batch

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

Недетерминированно группирует все входные тензоры.

Когда множество экземпляров этой операции запускаются одновременно с одним и тем же контейнером/shared_name на одном и том же устройстве, некоторые будут выводить тензоры нулевой формы, а другие будут выводить тензоры размером до max_batch_size.

Все тензоры в in_tensors объединяются вместе (поэтому, например, метки и объекты должны быть объединены в пакет с помощью одного экземпляра этой операции.

Каждый вызов пакета генерирует скаляр `id`, который будет использоваться для идентификации этого конкретного вызова при выполнении unbatch или его градиента.

Каждая операция, которая генерирует непустой пакет, также будет генерировать непустой тензор пакетного_индекса, который представляет собой матрицу [K, 3], где каждая строка содержит идентификатор вызова, начало и длину элементов каждого набора тензоров, присутствующих в пакетные_тензоры.

Пакетные тензоры объединяются по первому измерению, и все тензоры в in_tensors должны иметь первое измерение одинакового размера.

in_tensors: тензоры для пакетной обработки. num_batch_threads: количество потоков планирования для обработки пакетов работы. Определяет количество пакетов, обрабатываемых параллельно. max_batch_size: размеры пакетов никогда не будут больше этого значения. Batch_timeout_micros: Максимальное количество микросекунд ожидания перед выводом неполного пакета. allow_batch_sizes: необязательный список разрешенных размеров пакетов. Если оставить пустым, ничего не делает. В противном случае предоставляется список размеров пакетов, в результате чего оператор дополняет пакеты до одного из этих размеров. Записи должны увеличиваться монотонно, а конечная запись должна быть равна max_batch_size. grad_timeout_micros: тайм-аут, используемый для градиента. См. раздел «Отключить пакетирование». Batched_tensors: либо пустые тензоры, либо пакет объединенных тензоров. пакетный_индекс: если out_tensors не пусто, имеется информация для его инвертирования. контейнер: управляет областью совместного использования этого пакета. id: всегда содержит скаляр с уникальным идентификатором для этого вызова Batch. shared_name: одновременное выполнение экземпляров пакета на одном устройстве с одним и тем же контейнером и общим_имя будет группировать их элементы вместе. Если оставить пустым, имя операции будет использоваться в качестве общего имени. T: типы тензоров, подлежащих пакетной обработке.

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

сорт Пакетные.Параметры Дополнительные атрибуты для Batch

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

статические Batch.Options
разрешенныеBatchSizes (Список<Long> разрешенныеBatchSizes)
Вывод <Длинный>
Список< Вывод <?>>
статические Batch.Options
BatchingQueue (String BatchingQueue)
статические Batch.Options
контейнер (контейнер строк)
статический пакетный режим
create ( Область видимости , Iterable< Operand <?>> inTensors, Long numBatchThreads, Long maxBatchSize, LongatchTimeoutMicros, Long gradTimeoutMicros, Options... options)
Фабричный метод для создания класса, обертывающего новую пакетную операцию.
Вывод <Длинный>
статические Batch.Options
maxEnqueuedBatches (длинный maxEnqueuedBatches)
статические Batch.Options
общее имя (строка общее имя)

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

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

общедоступный статический Batch.Options разрешенныеBatchSizes (List<Long> разрешенныеBatchSizes)

публичный вывод <Long> пакетный индекс ()

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

общедоступный статический Batch.Options BattingQueue (String BatchingQueue)

общедоступный статический контейнер Batch.Options (контейнер String)

общедоступное статическое пакетное создание (область области действия , Iterable< Operand <?>> inTensors, Long numBatchThreads, Long maxBatchSize, LongatchTimeoutMicros, Long gradTimeoutMicros, Options... options)

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

Параметры
объем текущий объем
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр Batch

публичный выходной идентификатор <Long> ()

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

public static Batch.OptionssharedName ( StringsharedName)