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输出网络中间层结果的主要内容,如果未能解决你的问题,请参考以下文章