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

!48630 【OPS】fix the parameter dimension of operator roll must be 0 

Merge pull request !48630 from maoyaomin/roll_bug_mym
This commit is contained in:
i-robot 2023-02-10 01:42:14 +00:00 committed by Gitee
parent 9bb62cb742 027b708652
commit 70bf3ddea5
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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/python/mindspore/ops/operations/math_ops.py
View File

@ -7929,5 +7929,5 @@ class Roll(Primitive):
validator.check_equal_int(len(axis), 1, "shift size", self.name)
validator.check_equal_int(axis[0], 0, "axis", self.name)
elif isinstance(shift, int) and isinstance(axis, int):
validator.check_equal_int(axis, 0, "axis", self.name)
validator.check_is_int(axis, "axis", self.name)
self.init_prim_io_names(inputs=['input_x'], outputs=['output'])