[incubator-mxnet] branch master updated: fixes the incorrect retrieval of the batch size in the RNN forward pass. (#7385)

[jxie]

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>'].