diff --git a/mindspore/nn/layer/timedistributed.py b/mindspore/nn/layer/timedistributed.py index 5bfd3906ea..8f323ba1d9 100644 --- a/mindspore/nn/layer/timedistributed.py +++ b/mindspore/nn/layer/timedistributed.py @@ -41,7 +41,7 @@ def _check_expand_dims_axis(time_axis, ndim): def _generate_perm(axis_a, axis_b, length): perm = tuple(range(length)) axis_a, axis_b = (axis_a, axis_b) if axis_a < axis_b else (axis_b, axis_a) - return perm[:axis_a] + perm[axis_a + 1: axis_b + 1] + (perm[axis_a],) + perm[axis_b + 1:] + return perm[:axis_a] + (perm[axis_b],) + perm[axis_a: axis_b] + perm[axis_b + 1:] @constexpr