tf.parallel_stack

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

tf.parallel_stack(
    values, name='parallel_stack'
)

Требует, чтобы форма входных данных была известна на этапе построения графа.

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

Например:

x = tf.constant([1, 4])
y = tf.constant([2, 5])
z = tf.constant([3, 6])
tf.parallel_stack([x, y, z])  # [[1, 4], [2, 5], [3, 6]]

Разница между stack и parallel_stack заключается в том, что stack требует, чтобы все входные данные были вычислены до начала операции, но не требует, чтобы формы входных данных были известны во время построения графа.

parallel_stack будет копировать части входных данных в выходные данные по мере их доступности, в некоторых ситуациях это может обеспечить преимущество в производительности.

В отличие от stack, parallel_stack НЕ поддерживает обратное распространение.

Это противоположность unstack. Аналог в numpy:

tf.parallel_stack([x, y, z]) = np.asarray([x, y, z])