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.