публичный финальный класс TPUpartitionedOutputV2
Операция, которая демультиплексирует тензор, который будет сегментирован XLA в список секционированных
выходные данные вне вычислений XLA. Поддерживает шардинг ND.
Публичные методы
статический <T> TPUpartitionedOutputV2 <T> | create (область области действия , входные данные операнда <T>, Long numSplits, List<Long> partsDims) Фабричный метод для создания класса, обертывающего новую операцию TPUPartitionedOutputV2. |
Итератор< Операнд <T>> | итератор () |
Список< Выход <T>> | выход () Список разделенных выходных данных одинаковой формы. |
Унаследованные методы
Публичные методы
public static TPUPartitionedOutputV2 <T> create (область области действия , входные данные операнда <T>, Long numSplits, List<Long> partsDims)
Фабричный метод для создания класса, обертывающего новую операцию TPUPartitionedOutputV2.
Параметры
объем | текущий объем |
---|---|
входы | Тензор, который представляет полную форму разделенных тензоров. |
разделDims | Список целых чисел, описывающих, как разделено каждое измерение. Пустота указывает на то, что входные данные реплицируются. |
Возврат
- новый экземпляр TPUpartitionedOutputV2