tf.keras.config.set_backend

Перезагрузить бэкэнд (и пакет Keras).

tf.keras.config.set_backend(
    backend
)

Пример:

keras.config.set_backend("jax")

⚠️ ПРЕДУПРЕЖДЕНИЕ ⚠️: Использование этой функции опасно и требует осторожности. Изменение бэкэнда НЕ преобразует тип уже созданных объектов. Таким образом, любые слои/тензоры и т. д., уже созданные, больше не будут пригодны к использованию без ошибок. Настоятельно рекомендуется не хранить никакие экземпляры объектов, созданных с помощью Keras, до вызова set_backend().

Это включает в себя любые функции или экземпляры классов, использующие любые функции Keras. Весь такой код необходимо повторно выполнить после вызова set_backend().