gpt4 book ai didi

neural-network - 全连接 1 (fc1) 层的输入展平值来自哪里(MNIST 示例)

转载 作者:行者123 更新时间:2023-12-05 03:51:50 25 4
gpt4 key购买 nike

<分区>

这是 Pytorch 在其 github 上的示例目录中的一些卷积神经网络示例代码: https://github.com/pytorch/examples/blob/master/mnist/main.py

class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.conv1 = nn.Conv2d(1, 32, 3, 1)
self.conv2 = nn.Conv2d(32, 64, 3, 1)
self.dropout1 = nn.Dropout2d(0.25)
self.dropout2 = nn.Dropout2d(0.5)
self.fc1 = nn.Linear(9216, 128)
self.fc2 = nn.Linear(128, 10)

如果我理解这一点,我们需要先将最后一个卷积层的输出展平,然后才能将其传递给线性层 (fc1)。所以,看这段代码,我们看到第一个全连接层的输入是:9216。

这个数字 (9216) 来自哪里?

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com