PyTorch输出网络中间层结果

Posted 2023-01-14 算法与编程之美

问题

在研究网络某些行为特征的时候，除了需要网络最终的结果外，还需要获得网络某些中间层的输出。本章将介绍PyTorch中如何获得网络中间层的输出结果。

方法

只需要在定义网络的forward()中，保存并返回需要的层的结果即可，代码如下所示：

import torch
import torch.nn.functional as F
from torch import nn

\'\'\'输出网络中间层结果
\'\'\'
class Net(nn.Module):
    def