Torch LSTMCell Method
问题 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