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

fix the parameter dimension of operator roll must be 0

This commit is contained in:
maoyaomin 2023-02-09 15:41:36 +08:00
parent 32f8bf560a
commit 027b708652
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'])