!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:
commit
0ebb9dbdda
|
@ -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` 的元素不是非负的整数。
|
||||
|
|
|
@ -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]
|
||||
|
|
Loading…
Reference in New Issue