!39685 Update validation and doc for the para paddings for SpaceToBatchND

Merge pull request !39685 from zichun_ye/op_dev
This commit is contained in:
i-robot 2022-08-05 03:15:15 +00:00 committed by Gitee
commit 0ebb9dbdda
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 3 deletions

View File

@ -35,6 +35,6 @@ mindspore.ops.SpaceToBatchND
- **TypeError** - 如果 `paddings` 不是 list 或者 tuple。
- **ValueError** - 如果当 `block_shape` 为 list 或 tuple `block_shape` 不是一维。
- **ValueError** - 如果 Ascend 平台上 `block_shape` 长度不是2。
- **ValueError** - 如果 `paddings` 的形状不是 (2, M), 其中 M 为 `block_shape` 的长度。
- **ValueError** - 如果 `paddings` 的形状不是 (M, 2), 其中 M 为 `block_shape` 的长度。
- **ValueError** - 如果 `block_shape` 的元素不是大于一的整数。
- **ValueError** - 如果 `paddings` 的元素不是非负的整数。

View File

@ -5547,7 +5547,7 @@ class SpaceToBatchND(PrimitiveWithInfer):
TypeError: If `paddings` is neither list nor tuple.
ValueError: If `block_shape` is not one dimensional when `block_shape` is a list or tuple.
ValueError: If the length of `block_shape` is not 2 on Ascend.
ValueError: If shape of `paddings` is not (2, M), where M is the length of `block_shape`.
ValueError: If shape of `paddings` is not (M, 2), where M is the length of `block_shape`.
ValueError: If the element of `block_shape` is not an integer larger than 1.
ValueError: If the element of `paddings` is not an integer larger than 0.
@ -5571,7 +5571,6 @@ class SpaceToBatchND(PrimitiveWithInfer):
def __init__(self, block_shape, paddings):
"""Initialize SpaceToBatchND"""
validator.check_value_type('paddings type', paddings, [list, tuple], self.name)
validator.check('paddings length', len(paddings), 'default value', 2, Rel.EQ, self.name)
if isinstance(block_shape, int):
block_shape = (block_shape,) * np.array(paddings).shape[0]