This is an automated email from the ASF dual-hosted git repository. jxie pushed a commit to branch master in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push: new 1a617fa fixes the incorrect retrieval of the batch size in the RNN forward pass. (#7385) 1a617fa is described below commit 1a617fadc3cf612f964722e09571e1b80ad16f68 Author: Peter Schneider <pes...@users.noreply.github.com> AuthorDate: Tue Aug 8 19:30:03 2017 +0200 fixes the incorrect retrieval of the batch size in the RNN forward pass. (#7385) --- python/mxnet/gluon/rnn/rnn_layer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mxnet/gluon/rnn/rnn_layer.py b/python/mxnet/gluon/rnn/rnn_layer.py index d567369..deb6898 100644 --- a/python/mxnet/gluon/rnn/rnn_layer.py +++ b/python/mxnet/gluon/rnn/rnn_layer.py @@ -154,7 +154,7 @@ class _RNNLayer(Block): def forward(self, inputs, states): if isinstance(states, ndarray.NDArray): states = [states] - batch_size = states[0].shape[self._layout.find('N')] + batch_size = inputs.shape[self._layout.find('N')] for state, info in zip(states, self.state_info(batch_size)): if state.shape != info['shape']: raise ValueError( -- To stop receiving notification emails like this one, please contact ['"comm...@mxnet.apache.org" <comm...@mxnet.apache.org>'].