当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch register_module_forward_hook用法及代码示例


本文简要介绍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 注册的特定模块钩子之前执行。

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.modules.module.register_module_forward_hook。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。