Pytorch Pytork：对模型特定层的推断

我正在构建一个灵活的模块，它实现了Monte Carlo辍学，用于在任何类型的模型上进行贝叶斯推理。 （通过在测试期间向fc层添加压降实现MC压降，并多次推断以获得输出分布）

为了节省时间，我想将模型分为两部分：

未实现辍学，因此只运行一次

因此，在第一个退出（第一个fc层）之后的层将运行多次

我不能使用

nn.Sequential（）

您知道如何仅在特定层上进行推断而不更改模型本身或
forward（）
代码吗