Get layer's output from nn.Sequential

2

You could use forward hooks as described here.

1 Like