ReadVariableXlaSplitND

публичный финальный класс ReadVariableXlaSplitND

Разбивает входной тензор переменной ресурса по всем измерениям.

Операция, которая разбивает входной тензор переменной ресурса на основе заданного атрибута num_splits, дополнительно дополняет срезы и возвращает срезы. Срезы возвращаются в порядке строк.

Эта операция может быть сгенерирована через мост TPU.

Например, для тензора `input`:

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
 
`num_splits`:
[2, 2]
 
и `paddings`:
[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

общедоступный итератор < Операнд <T>> итератор ()

публичный список< Вывод <T>> выходы ()

общедоступные статические дополнения ReadVariableXlaSplitND.Options (дополнения List<Long>)

Параметры
прокладки Необязательный список правых заполнений для каждого измерения входного тензора, которые необходимо применить перед разделением. Это можно использовать для того, чтобы сделать размерность делимой без остатка.