tf.keras.ops.fori_loop
Реализация цикла for.
tf.keras.ops.fori_loop(
lower, upper, body_fun, init_val
)
| Аргументы | |
|---|---|
lower | Начальное значение переменной цикла. |
upper | Верхняя граница переменной цикла. |
body_fun | Вызываемый объект, представляющий тело цикла. Должен принимать два аргумента: переменную цикла и состояние цикла. Состояние цикла должно быть обновлено и возвращено этой функцией. |
init_val | Начальное значение состояния цикла. |
| Возвращаемые значения | |
|---|---|
| Конечное состояние после цикла. |
Пример:
lower = 0 upper = 10 body_fun = lambda i, s: (i + 1, s + i) init_val = 0 keras.ops.fori_loop(lower, upper, body_fun, init_val) 45
© 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/fori_loop
