публичный финальный класс 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>> вывод ()
Список разделенных входов, которые должны иметь одинаковую форму.
общедоступный статический TPPUpartitionedOutput.Options разделDim (длинный разделDim)
Параметры
разделDim | Целое число описывает, какое измерение разделено. |
---|