Функция concat
Функция concat объединяет несколько тензоров в один вдоль заданной оси. Первым параметром функция принимает список или кортеж тензоров, вторым параметром передается ось, вдоль которой происходит объединение. Третьим необязательным параметром можно передать имя операции.
Синтаксис
tf.concat(values, axis, [name])
Пример
Давайте объединим два одномерных тензора вдоль оси 0:
import tensorflow as tf
t1 = tf.constant([1, 2, 3])
t2 = tf.constant([4, 5, 6])
t = tf.concat([t1, t2], axis=0)
print(t)
Результат выполнения кода:
tf.Tensor([1 2 3 4 5 6], shape=(6,), dtype=int32)
Пример
Давайте объединим два двумерных тензора вдоль оси 0 (по строкам):
import tensorflow as tf
t1 = tf.constant([[1, 2, 3]])
t2 = tf.constant([[4, 5, 6]])
t = tf.concat([t1, t2], axis=0)
print(t)
Результат выполнения кода:
tf.Tensor(
[[1 2 3]
[4 5 6]], shape=(2, 3), dtype=int32)
Пример
Давайте объединим два двумерных тензора вдоль оси 1 (по столбцам):
import tensorflow as tf
t1 = tf.constant([[1, 2], [3, 4]])
t2 = tf.constant([[5, 6], [7, 8]])
t = tf.concat([t1, t2], axis=1)
print(t)
Результат выполнения кода:
tf.Tensor(
[[1 2 5 6]
[3 4 7 8]], shape=(2, 4), dtype=int32)
