Autograph пытается превратить Python if/while/for в tf.cond/tf.while_loop, если условия тензорные; иначе остаётся Python-side на trace.
Разбор
- Тензорные условия → графовые ветки.
- Python bool от
.numpy() ломает граф.
- Сложные side effects лучше выносить.
- Документация Autograph — must-read для custom loops.
Итог
Autograph переводит Python-контроль в графовые конструкции.