TPUPartitionedOutput

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

Операция, которая демультиплексирует тензор, который будет сегментирован XLA в список секционированных

выходные данные вне вычислений XLA.

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

сорт ТПУПартиционедаутпут.Параметры Дополнительные атрибуты для TPUPartitionedOutput

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

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

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

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

public static TPUPartitionedOutput <T> create (область области действия , входные данные операнда <T>, длинные numSplits, параметры... параметры)

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

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

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

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

Список разделенных входов, которые должны иметь одинаковую форму.

общедоступный статический TPPUpartitionedOutput.Options разделDim (длинный разделDim)

Параметры
разделDim Целое число описывает, какое измерение разделено.