问题
self.conv1 = nn.Conv2d(num_inputs, 32, 3, stride=2, padding=1)
self.conv2 = nn.Conv2d(32, 32, 3, stride=2, padding=1)
self.conv3 = nn.Conv2d(32, 32, 3, stride=2, padding=1)
self.conv4 = nn.Conv2d(32, 32, 3, stride=2, padding=1)
self.lstm = nn.LSTMCell(32 * 3 * 3, 256)
Can somebody provide the formula of the calculation lstm input (32 * 3 * 3). Where is this 3 comes from.
来源:https://stackoverflow.com/questions/62724776/torch-lstmcell-method