对于nn.Sequential结构,要想获取中间网络层输出,可以使用循环遍历的方式得到。
示例
import torch
import torch.nn as nn
model = nn.Sequential(
nn.Conv2d(3, 9, 1, 1, 0, bias=False<
本文介绍如何在PyTorch中通过遍历Sequential模型的每一层来捕获中间层的输出,示例代码展示了如何获取ReLU层的输出,并提供了输出张量的shape信息。
对于nn.Sequential结构,要想获取中间网络层输出,可以使用循环遍历的方式得到。
示例
import torch
import torch.nn as nn
model = nn.Sequential(
nn.Conv2d(3, 9, 1, 1, 0, bias=False<
9450
1825

被折叠的 条评论
为什么被折叠?