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>)
Параметры
прокладки | Необязательный список правых полей для каждого измерения, которые нужно удалить из окончательного объединенного тензора. Эти отступы не должны превышать размерность объединенного результата до удаления отступов. |
---|