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.