本文簡要介紹python語言中 torch.nn.Module.register_forward_pre_hook
的用法。
用法:
register_forward_pre_hook(hook)
一個句柄,可用於通過調用
handle.remove()
刪除添加的鉤子torch.utils.hooks.RemovableHandle
在模塊上注冊一個前向預掛鉤。
每次調用
forward()
之前都會調用該鉤子。它應該具有以下簽名:hook(module, input) -> None or modified input
輸入僅包含給模塊的位置參數。關鍵字參數不會傳遞給鉤子,隻會傳遞給
forward
。鉤子可以修改輸入。用戶可以在鉤子中返回一個元組或單個修改值。如果返回單個值,我們會將值包裝到一個元組中(除非該值已經是一個元組)。
返回:
返回類型:
相關用法
- Python PyTorch Module.register_forward_hook用法及代碼示例
- Python PyTorch Module.register_full_backward_hook用法及代碼示例
- Python PyTorch Module.register_buffer用法及代碼示例
- Python PyTorch Module.buffers用法及代碼示例
- Python PyTorch Module.named_modules用法及代碼示例
- Python PyTorch Module.parameters用法及代碼示例
- Python PyTorch Module.named_parameters用法及代碼示例
- Python PyTorch Module.state_dict用法及代碼示例
- Python PyTorch Module.named_children用法及代碼示例
- Python PyTorch Module.modules用法及代碼示例
- Python PyTorch Module.apply用法及代碼示例
- Python PyTorch Module.to用法及代碼示例
- Python PyTorch Module.named_buffers用法及代碼示例
- Python PyTorch ModuleList用法及代碼示例
- Python PyTorch Module用法及代碼示例
- Python PyTorch ModuleDict用法及代碼示例
- Python PyTorch MaxUnpool3d用法及代碼示例
- Python PyTorch MultiStepLR用法及代碼示例
- Python PyTorch MaxPool1d用法及代碼示例
- Python PyTorch MetaInferGroupedPooledEmbeddingsLookup.state_dict用法及代碼示例
- Python PyTorch MetaInferGroupedEmbeddingsLookup.named_buffers用法及代碼示例
- Python PyTorch MultiLabelMarginLoss用法及代碼示例
- Python PyTorch MultiplicativeLR用法及代碼示例
- Python PyTorch MixtureSameFamily用法及代碼示例
- Python PyTorch MultiheadAttention用法及代碼示例
注:本文由純淨天空篩選整理自pytorch.org大神的英文原創作品 torch.nn.Module.register_forward_pre_hook。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。