torch.nn.modules.module.register_module_forward_pre_hook

torch.nn.modules.module.register_module_forward_pre_hook(hook)

註冊一個所有模組通用的 forward pre-hook。

警告

這會給 nn.module 模組增加全域性狀態，它僅用於除錯/效能分析目的。

每次呼叫 forward() 之前都會呼叫該 hook。它應該具有以下 signature

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

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

該 hook 優先於使用 register_forward_pre_hook 註冊的特定模組 hook。

返回

一個 handle，可以透過呼叫 handle.remove() 來移除新增的 hook

返回型別

torch.utils.hooks.RemovableHandle