本文简要介绍python语言中 torch.nn.modules.module.register_module_forward_pre_hook
的用法。
用法:
torch.nn.modules.module.register_module_forward_pre_hook(hook)
一个句柄,可用于通过调用
handle.remove()
删除添加的钩子torch.utils.hooks.RemovableHandle
注册所有模块通用的前向预挂钩。
警告
这会将全局状态添加到
nn.module
模块,并且仅用于调试/分析目的。每次调用
forward()
之前都会调用该钩子。它应该具有以下签名:hook(module, input) -> None or modified input
输入仅包含给模块的位置参数。关键字参数不会传递给钩子,只会传递给
forward
。钩子可以修改输入。用户可以在钩子中返回一个元组或单个修改值。如果返回单个值,我们会将值包装到一个元组中(除非该值已经是一个元组)。此挂钩优先于使用
register_forward_pre_hook
注册的特定模块挂钩。
返回:
返回类型:
相关用法
- Python PyTorch register_module_forward_hook用法及代码示例
- Python PyTorch register_module_full_backward_hook用法及代码示例
- Python PyTorch register_kl用法及代码示例
- Python PyTorch register_parametrization用法及代码示例
- Python PyTorch renorm用法及代码示例
- Python PyTorch reshape用法及代码示例
- Python PyTorch real用法及代码示例
- Python PyTorch repeat_interleave用法及代码示例
- Python PyTorch remove用法及代码示例
- Python PyTorch read_vec_flt_ark用法及代码示例
- Python PyTorch read_vec_int_ark用法及代码示例
- Python PyTorch resolve_neg用法及代码示例
- Python PyTorch remainder用法及代码示例
- Python PyTorch remote用法及代码示例
- Python PyTorch remove_spectral_norm用法及代码示例
- Python PyTorch record用法及代码示例
- Python PyTorch remove_weight_norm用法及代码示例
- Python PyTorch retinanet_resnet50_fpn用法及代码示例
- Python PyTorch read_vec_flt_scp用法及代码示例
- Python PyTorch resolve_conj用法及代码示例
- Python PyTorch reciprocal用法及代码示例
- Python PyTorch result_type用法及代码示例
- Python PyTorch replace_pattern用法及代码示例
- Python PyTorch read_mat_scp用法及代码示例
- Python PyTorch read_mat_ark用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.modules.module.register_module_forward_pre_hook。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。