Extract intermediate layer

[ptrblck]

You could register a forward hook to the desired layer as described here.