fix.

apache · Aug 7, 2018 · cf356c2 · cf356c2
1 parent debfba5
commit cf356c2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/python/mxnet/gluon/contrib/rnn/rnn_cell.py b/python/mxnet/gluon/contrib/rnn/rnn_cell.py
@@ -410,9 +410,7 @@ def unroll(cell, inputs, begin_state, drop_inputs=0, drop_outputs=0,
     inputs, axis, F, _ = _contrib_format_sequence(inputs, layout)
     if axis != 0:
         axes = list(range(len(layout)))
-        tmp = axes[0]
-        axes[0] = axes[axis]
-        axes[axis] = tmp
+        axes[0], axes[axis] = axes[axis], axes[0]
         inputs = F.transpose(inputs, axes=axes)
     states = begin_state