You could use forward hooks to get the forward activation. I’ve created a small example in this post.