tf.contrib.tpu.while_loop
Создаёт цикл обучения для TPUs.
tf.contrib.tpu.while_loop(
condition, body, inputs=None, infeed_queue=None, name=None
)
Набор тензоров, переносимых по циклу, соответствует inputs. Оба condition и body принимают текущее значение тензоров, переносимых по циклу. 'body' также принимает кортеж из входных данных из очереди infeed_queue, если infeed_queue не равно None. condition должно возвращать единственное булевое значение, определяющее, продолжается ли итерация. body должно возвращать обновлённый список значений для тензоров, переносимых по циклу.
| Аргументы | |
|---|---|
condition | функция Python, которая строит условие цикла. |
body | функция Python, которая строит тело цикла. |
inputs | список начальных значений, передаваемых в цикл обучения, или None (эквивалентно пустому списку). |
infeed_queue | если не равно None, очередь infeed, из которой нужно добавить кортеж аргументов в качестве входных данных для условия. |
name | (Устарело) Ничего не делает. |
| Возвращаемое значение | |
|---|---|
| Конечные значения тензоров, переносимых по циклу. |
| Исключения | |
|---|---|
TypeError | если у body или condition неправильная сигнатура. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/tpu/while_loop
