questions about the initiate of the contextual module

[munanning]

dear fyu， where can I find how the “identity” isdefined？ I want to rebuild it using Keras, and I wish I can use your code for reference