Replies: 1 comment
-
|
Hey @sglucas, take a look at the Extracting intermediate values guide. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
May I ask how to obtain the output of each layer and the activation values when i use JAX with flax? I hope you could provide a simple example.
Best,
Lucas
Beta Was this translation helpful? Give feedback.
All reactions