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