публичный финальный класс 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 | Целое число описывает, какое измерение разделено. |
|---|