PyTorch Forums
How to modify and rewrite the activation output of a layer before applying the output to next layer?
ptrblck
November 28, 2020, 7:56am
2
You could use forward hooks as described
here
and manipulate the
output
inplace in the hook.
show post in topic