torch.nn.modules.module.register_module_forward_pre_hook¶
- torch.nn.modules.module.register_module_forward_pre_hook(hook)[源代码]¶
注册所有模块共有的前向预钩子。
警告
这会向 nn.module 模块添加全局状态,仅用于调试/性能分析目的。
每次在调用
forward()
之前,都会调用该钩子。它应具有以下签名hook(module, input) -> None or modified input
输入仅包含传递给模块的位置参数。关键字参数不会传递给钩子,只会传递给
forward
。钩子可以修改输入。用户可以在钩子中返回元组或单个修改后的值。如果返回单个值(除非该值已经是元组),我们将将其包装到元组中。此钩子优先于使用
register_forward_pre_hook
注册的特定模块钩子。- 返回值
一个句柄,可用于通过调用
handle.remove()
删除添加的钩子- 返回类型
torch.utils.hooks.RemovableHandle