Разбивает входной тензор по всем измерениям.
Операция, которая разрезает входной тензор на основе заданного атрибута num_splits, дополняет фрагменты (необязательно) и возвращает фрагменты. Срезы возвращаются в порядке строк.
Эта операция может быть сгенерирована через мост TPU.
Например, с тензором `input`:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[2, 2]
[1, 1]
[[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Вложенные классы
сорт | XlaSplitND.Параметры | Дополнительные атрибуты для XlaSplitND |
Публичные методы
статический <T> XlaSplitND <T> | create (Область действия , ввод операнда <T>, Long N, List<Long> numSplits, параметры... ) Фабричный метод для создания класса, обертывающего новую операцию XlaSplitND. |
Итератор< Операнд <T>> | итератор () |
Список< Выход <T>> | выходы () |
статический XlaSplitND.Options | отступы (отступы List<Long>) |
Унаследованные методы
Публичные методы
public static XlaSplitND <T> create (область действия , ввод операнда <T>, Long N, List<Long> numSplits, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию XlaSplitND.
Параметры
объем | текущий объем |
---|---|
вход | Введите тензор для разделения по всем измерениям. } out_arg { name: "outputs" описание: < |
numSplits | Количество способов разделения по измерению. Размеры фигуры должны делиться без остатка. |
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр XlaSplitND
общедоступные статические дополнения XlaSplitND.Options (дополнения List<Long>)
Параметры
прокладки | Необязательный список правых заполнений для каждого измерения входного тензора, которые необходимо применить перед разделением. Это можно использовать для того, чтобы сделать размерность делимой без остатка. |
---|