如何在Torch7中获取中间层的输出?

时间:2017-05-17 02:14:45

标签: torch

似乎当我使用nngraph模型时,很容易这样做。 如果我只使用普通转换器,那么我该怎么做呢?

1 个答案:

答案 0 :(得分:0)

定义模型时,例如带有nn.Sequential()的convnet并向其添加模块

net = nn.Sequential()
net:add(nn.SpatialConvolution(3,3,1,1)
net:add(...) -- add other modules

您可以访问具有net.modules[n]的模块(n是模块的索引,使用print(net)查看整个网络及其模块)。然后每个模块必须声明变量outputgradInput(模块相对于其输入的渐变),然后您可以使用

访问第n个中间层的输出
net.modules[n].output