I’m using forward hooks to look at the intermediate activations of a net as it goes through, works great.
I’d like to be able to correlate/separate these activations by the input, so I can see just the activations for a particulr input, even if multiple inputs are being processed at the same time or while the hooks are in place. In essence I’d like to be able to separate the data that the hooks give me into something like:
For input #1 → activations #1
input #2 → activations #2
As I understand it now, the hooks give me the activations but won’t tell me for which input they are.
Make sense? Thanks in advance