Torch LSTMCell Method

独自空忆成欢 提交于 2020-07-23 06:24:19

问题


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

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!