You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by jx...@apache.org on 2018/04/27 16:40:41 UTC

[incubator-mxnet] 02/02: Update rnn_layer.py

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

commit 32aff9b8acd07f9b6ab4de119a29b80fff39db43
Author: Sheng Zha <sz...@users.noreply.github.com>
AuthorDate: Thu Apr 26 22:54:19 2018 -0700

    Update rnn_layer.py
---
 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 0559445..a4150c9 100644
--- a/python/mxnet/gluon/rnn/rnn_layer.py
+++ b/python/mxnet/gluon/rnn/rnn_layer.py
@@ -91,7 +91,7 @@ class _RNNLayer(Block):
             s += ', bidirectional'
         s += ')'
         shape = self.i2h_weight[0].shape
-        mapping = '{0} -> {1}'.format(shape[1] if shape[1] else None, int(shape[0] / self._gates))
+        mapping = '{0} -> {1}'.format(shape[1] if shape[1] else None, shape[0] // self._gates)
         return s.format(name=self.__class__.__name__,
                         mapping=mapping,
                         **self.__dict__)

-- 
To stop receiving notification emails like this one, please contact
jxie@apache.org.