публичный финальный класс AssignVariableXlaConcatND
Concats входной тензор по всем измерениям.
Операция, которая объединяет, разрезает входной тензор на основе заданного атрибута num_splits, при необходимости удаляет заполнения и записывает объединенный тензор без дополнений в переменную ресурса.
Эта операция может быть сгенерирована через мост TPU.
Например, для тензора `input`:
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
[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>)
Параметры
| прокладки | Необязательный список правых полей для каждого измерения, которые нужно удалить из окончательного объединенного тензора. Эти отступы не должны превышать размерность объединенного результата до удаления отступов. |
|---|