tf.raw_ops.XlaConcatND

Конкатенация входного тензора по всем измерениям.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.raw_ops.XlaConcatND

tf.raw_ops.XlaConcatND(
    inputs, num_concats, paddings=[], name=None
)

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

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

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

[[0, 1],
 [4, 5]]
[[2, 3],
 [6, 7]]
[[8, 9],
 [12, 13]]
[[10, 11],
 [14, 15]]

num_splits:

[2, 2]

и paddings:

[1, 1]

ожидаемый outputs:

[[0, 1, 2],
 [4, 5, 6],
 [8, 9, 10]]

Аргументы
`inputs`	Список из как минимум 1 `Tensor` объектов с одинаковым типом. Сlices входного тензора в порядке следования строк для объединения по всем измерениям. Все входные данные должны иметь одинаковую форму. } out_arg { name: "output" description: <
`num_concats`	Список `ints`. Количество способов объединения по измерению.
`paddings`	Необязательный список `ints`. По умолчанию `[]`. Необязательный список правых заполнений по каждому измерению, которые нужно удалить из окончательного объединенного тензора. Эти заполнения не должны превышать размер измерения объединенного результата до удаления заполнений.
`name`	Имя операции (необязательно).

Возвращаемые значения
`Tensor`. Имеет тот же тип, что и `inputs`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/raw_ops/XlaConcatND