PyTorch输出网络中间层结果

Posted 算法与编程之美

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PyTorch输出网络中间层结果相关的知识,希望对你有一定的参考价值。

问题

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

方法

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

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

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

以上是关于PyTorch输出网络中间层结果的主要内容,如果未能解决你的问题,请参考以下文章

pytorch hook 钩子

Pytorch学习之梯度计算backward函数

取出预训练模型中间层的输出(pytorch)

神经网络简介(pytorch)

动手学pytorch-Batch Norm

pytorch层顺序会影响结果吗