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