Could you wrap the operation involving this parameter in a custom module and register the forward hook?
I assume you are performing some operation using this parameter in your forward method.
If so, you could also return/store the output of this operation.