本文简要介绍python语言中 torch.nn.modules.module.register_module_forward_hook
的用法。
用法:
torch.nn.modules.module.register_module_forward_hook(hook)
一个句柄,可用于通过调用
handle.remove()
删除添加的钩子torch.utils.hooks.RemovableHandle
为所有模块注册一个全局前向钩子
警告
这会将全局状态添加到
nn.module
模块,并且仅用于调试/分析目的。每次
forward()
计算输出后都会调用该钩子。它应该具有以下签名:hook(module, input, output) -> None or modified output
输入仅包含给模块的位置参数。关键字参数不会传递给钩子,只会传递给
forward
。钩子可以修改输出。它可以就地修改输入,但不会对转发产生影响,因为这是在调用forward()
之后调用的。该钩子将在使用
register_forward_hook
注册的特定模块钩子之前执行。
返回:
返回类型:
相关用法
- Python PyTorch register_module_forward_pre_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_hook。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。