How to get output parameter from ScriptModules

1

Hello everyone, I want to get output parameter from ScriptModules layer.

But

RuntimeError: register_forward_hook is not supported on ScriptModules

Is there any other way to achieve it?

thx you.