register_forward_hook 怎么用

register_forward_hook 是 PyTorch 中的一个函数，它可以让你在模型的前向传播过程中，在每一层的输出之后插入自己的处理代码。

使用方法如下：

import torch

# 定义一个 hook 函数，输入是当前层的输出，输出是处理后的输出
def my_hook(module, input, output):
    # 在这里处理 output
    # 例如，让输出乘以 2
    output = output * 2
    return output

# 建立一个网络，并在第一层之后插入 hook
model = torch.