tf.raw_ops.Pack

Упаковывает список N тензоров ранга R в один тензор ранга (R+1).

tf.raw_ops.Pack(
    values, axis=0, name=None
)

Упаковывает тензоры N в values в тензор с рангом на один выше, чем у каждого тензора в values, упаковывая их по размерности axis. Учитывая список тензоров формы (A, B, C);

если axis == 0, то тензор output будет иметь форму (N, A, B, C). Если axis == 1, то тензор output будет иметь форму (A, N, B, C). И т.д.

Например:

# 'x' is [1, 4]
# 'y' is [2, 5]
# 'z' is [3, 6]
pack([x, y, z]) => [[1, 4], [2, 5], [3, 6]]  # Pack along first dim.
pack([x, y, z], axis=1) => [[1, 2, 3], [4, 5, 6]]

Это противоположно unpack.