torch.nn.modules.module.register_module_forward_pre_hook

torch.nn.modules.module.register_module_forward_pre_hook(hook)[source][source]

向所有模块注册通用的前向钩子

警告

这将为 nn.module 模块添加全局状态，仅用于调试/性能分析目的。

每当调用 forward() 之前，都会调用此钩子。它应该具有以下签名：

hook(module, input) -> None or modified input

输入仅包含传递给模块的位置参数。关键字参数不会传递给钩子，只会传递给 forward 。钩子可以修改输入。用户可以在钩子中返回一个元组或单个修改后的值。如果返回单个值（除非该值已经是元组），我们将将其包装成元组。

此钩子的优先级高于使用 register_forward_pre_hook 注册的特定模块钩子。

返回值:

一个可以用来通过调用 handle.remove() 移除已添加钩子的句柄。

返回类型:

torch.utils.hooks.RemovableHandle