tf.raw_ops.AssignVariableXlaConcatND

Объединяет тензор входных данных по всем измерениям.

tf.raw_ops.AssignVariableXlaConcatND(
    resource, 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]]