tf.keras.ops.unstack

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

tf.keras.ops.unstack(
    x, num=None, axis=0
)

Аргументы
`x`	Входной тензор.
`num`	Длина оси измерения. Автоматически определяется, если `None`.
`axis`	Ось, вдоль которой происходит распаковка.

Возвращаемое значение
Список тензоров, распакованных вдоль заданной оси.

x = keras.ops.array([[1, 2], [3, 4]])
keras.ops.unstack(x, axis=0)
[array([1, 2]), array([3, 4])]

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/unstack