ops模块API文档修改

This commit is contained in:
yangxixin 2022-12-02 10:47:38 +08:00
parent f7b2d406d1
commit f08ac79156
2 changed files with 3 additions and 4 deletions

View File

@ -517,8 +517,8 @@ class Col2Im(Primitive):
by summing the overlapping values.
.. math::
L = \prod_d \left\lfloor\frac{\text{output\_size}[d] + 2 \times \text{padding}[d] %
- \text{dilation}[d] \times (\text{kernel\_size}[d] - 1) - 1}{\text{stride}[d]} + 1\right\rfloor,
L = \prod_d \left\lfloor\frac{\text{output_size}[d] + 2 \times \text{padding}[d] %
- \text{dilation}[d] \times (\text{kernel_size}[d] - 1) - 1}{\text{stride}[d]} + 1\right\rfloor,
where :math:`d` is over all spatial dimensions.

View File

@ -1112,9 +1112,8 @@ class CombinedNonMaxSuppression(Primitive):
TypeError: If the dtype of `max_output_size_per_class` and `max_total_size` are not int32.
ValueError: If `boxes` is not 4D.
ValueError: If `max_output_size_per_class`, `max_total_size`, `iou_threshold` and `score threshold` are not 0D.
ValueError: If shape[0] of `boxes` is not same with shape[0] of `scores`.
ValueError: If `scores` is not 3D.
ValueError: If shape[1] of `boxes` is not same with shape[1] of the `scores`.
ValueError: If shape[0] or shape[1] of `boxes` is not same with that of the `scores`.
ValueError: If shape[2] of `boxes` is not same with shape[2] of `scores` or 1
ValueError: If `max_total_size` < 0.
ValueError: If `max_output_size_per_class` < 0.