如何获取 PyTorch 中 nn.Sequential 模块执行过程中的中间步骤输出？

在PyTorch中，`nn.Sequential`是一个容器模块，用于按照给定的顺序连接多个模块。如果你想在`Sequential`执行过程中获取中间步骤的输出，你可以通过将需要记录输出的模块添加到序列，并在它们之后设置一个`torch.nn.ModuleList`或者自定义的类，该类可以存储并返回这些中间输出。

例如：

class IntermediateOutputs(nn.Module):
    def __init__(self):
        super(IntermediateOutputs, self).__init__()
        self.outputs = []

    def forward(self, x):
        for module in self.sequential:
            x = module(x)
            self.outputs.append(x)  # 在每个模块之后，将输出存入outputs列表
        return x

# 创建一个Sequential模型
model = nn.Sequential(
    ...,
    # 添加需要监控输出的模块
    MyCustomModule(),
    ...,
)

# 创建IntermediateOutputs实例，将Sequential传递进去
intermediate_output_handler = IntermediateOutputs()
intermediate_output_handler.sequential = model

# 运行模型并访问中间输出
output = intermediate_output_handler(input)

在这个例子中，`intermediate_output_handler.outputs`将包含`MyCustomModule`之后的所有中间层输出。每次模型前向传播，这个列表就会更新。