use a diff default layout.

apache · Aug 27, 2018 · f7b5799 · f7b5799
1 parent 904404e
commit f7b5799
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mxnet/gluon/contrib/rnn/rnn_cell.py b/python/mxnet/gluon/contrib/rnn/rnn_cell.py
@@ -342,7 +342,7 @@ def _contrib_format_sequence(inputs, layout, in_layout=None):
 
 
 def unroll(cell, inputs, begin_state, drop_inputs=0, drop_outputs=0,
-           layout='NTC', valid_length=None):
+           layout='TNC', valid_length=None):
     """Unrolls an RNN cell across time steps.
 
     Currently, 'TNC' is a preferred layout. unroll on the input of this layout