В 5 раз медленнее, чем Pytorch (если вы видите какую -либо легкую оптимизацию, пожалуйста, дайте мне знать.)
Оптимизация с помощью tf.function
TF.Function выглядит так сложно использовать с моделями, имеющими разные входные формы на итерацию, поскольку TF пересматривает график для каждого входа. Насколько я вижу из официальных примеров, обходные пути используют преимущества использования нетерпеливого режима. (Поправьте меня, если я ошибаюсь.)