p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update mindspore/nn/layer/timedistributed.py. 

Fix: reshape with aix bug.
This commit is contained in:
dinglongwei 2021-03-03 14:17:13 +08:00 committed by Gitee
parent 885177769a
commit 640487a974
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/layer/timedistributed.py
View File

@ -41,7 +41,7 @@ def _check_expand_dims_axis(time_axis, ndim):
def _generate_perm(axis_a, axis_b, length): def _generate_perm(axis_a, axis_b, length):
perm = tuple(range(length)) perm = tuple(range(length))
axis_a, axis_b = (axis_a, axis_b) if axis_a < axis_b else (axis_b, axis_a) 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 @constexpr