Table Of Contents
Table Of Contents

Block.register_forward_hook

Block.register_forward_hook(hook)[source]

Registers a forward hook on the block.

The hook function is called immediately after forward(). It should not modify the input or output.

Parameters:hook (callable) – The forward hook function of form hook(block, input, output) -> None.
Returns:
Return type:mxnet.gluon.utils.HookHandle