tf.unstack

Распаковывает данную размерность тензора ранга R в тензоры ранга (R-1).

tf.unstack(
    value, num=None, axis=0, name='unstack'
)

Распаковывает num тензоры из value, разбивая его вдоль размерности axis. Если num не указано (по умолчанию), оно выводится из формы value. Если value.shape[axis] неизвестно, поднимается ValueError.

Например, задан тензор формы (A, B, C, D);

Если axis == 0, то i-й тензор в output — срез value[i, :, :, :], и каждый тензор в output будет иметь форму (B, C, D). (Обратите внимание, что размерность, по которой происходит распаковка, исчезает, в отличие от split).

Если axis == 1, то i-й тензор в output — срез value[:, i, :, :], и каждый тензор в output будет иметь форму (A, C, D). И т.д.

Это обратная операция к stack.