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