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.