torch.nn.modules.module.register_module_forward_pre_hook#

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

為所有模組註冊一個通用的前向預鉤子。

警告

此函式會向 nn.module 模組新增全域性狀態，僅用於除錯/剖析目的。

每次呼叫 forward() 之前都會呼叫該鉤子。它應該具有以下簽名：

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

輸入僅包含傳遞給模組的位置引數。關鍵字引數不會傳遞給鉤子，只會傳遞給 forward。鉤子可以修改輸入。使用者可以在鉤子中返回一個元組或單個修改後的值。如果返回單個值（除非該值已經是元組），我們將把該值包裝成一個元組。

此鉤子具有高於使用 register_forward_pre_hook 註冊的特定模組鉤子的優先順序。

文件