AssignVariableXlaConcatND

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

Concats входной тензор по всем измерениям.

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

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

Например, с тензором `input`:

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
`num_splits`:
[2, 2]
и `отступы`:
[1, 1]
ожидаемые «результаты»:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]

Вложенные классы

сорт AssignVariableXlaConcatND.Options Дополнительные атрибуты для AssignVariableXlaConcatND

Публичные методы

статический <T> AssignVariableXlaConcatND
create ( Область действия , ресурс Operand <?>, Iterable< входы операнда <T>>, List<Long> numConcats, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию AssignVariableXlaConcatND.
статический AssignVariableXlaConcatND.Options
отступы (отступы List<Long>)

Унаследованные методы

Публичные методы

public static AssignVariableXlaConcatND create ( область области действия, ресурс Operand <?>, входные данные Iterable< Operand <T>>, List<Long> numConcats, параметры... параметры)

Фабричный метод для создания класса, обертывающего новую операцию AssignVariableXlaConcatND.

Параметры
объем текущий объем
ресурс Ресурсная переменная для объединенных входных тензоров по всем измерениям. } in_arg { name: "inputs" описание: <
numConcats Количество способов слияния для каждого измерения.
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр AssignVariableXlaConcatND

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

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