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(module, input) -> None or modified input

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

此 hook 的優先級高於使用 register_forward_pre_hook 註冊的特定模組 hooks。

回傳值

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

回傳類型

torch.utils.hooks.RemovableHandle