tf.keras.ops.while_loop

Реализация цикла while.

tf.keras.ops.while_loop(
    cond, body, loop_vars, maximum_iterations=None
)

Аргументы
`cond`	Вызываемый объект, представляющий условие завершения цикла. Должен принимать в качестве аргумента структуру вида `loop_vars`. Если `loop_vars` является кортежем или списком, каждый элемент `loop_vars` будет передан вызываемому объекту позиционно.
`body`	Вызываемый объект, представляющий тело цикла. Должен принимать в качестве аргумента структуру вида `loop_vars` и возвращать обновлённое значение с той же структурой. Если `loop_vars` является кортежем или списком, каждый элемент `loop_vars` будет передан вызываемому объекту позиционно.
`loop_vars`	Произвольная вложенная структура состояний тензоров, сохраняемая между итерациями цикла.
`maximum_iterations`	Необязательное максимальное число итераций цикла while. Если задано, результат `cond` объединяется логическим И с дополнительным условием, гарантирующим, что число выполненных итераций не превышает `maximum_iterations`.

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

Примеры:

i = 0
cond = lambda i: i < 10
body = lambda i: i + 1
keras.ops.while_loop(cond, body, i)
10

x, y = 0, 1
cond = lambda x, y: x < 10
body = lambda x, y: (x + 1, y + 1)
keras.ops.while_loop(cond, body, (x, y))
10, 11

© 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/while_loop