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 注册的特定模块钩子。