I had to use keras backend’s switch function for the automatic differentiation to work, but basically yes.
I had to use keras backend’s switch function for the automatic differentiation to work, but basically yes.