Python PyTorch Module.register_forward_pre_hook用法及代码示例

本文简要介绍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 。钩子可以修改输入。用户可以在钩子中返回一个元组或单个修改值。如果返回单个值，我们会将值包装到一个元组中(除非该值已经是一个元组)。