add the picture of calculation for UnsortedSegmentxxx.

This commit is contained in:
wangshuide2020 2021-09-07 20:17:17 +08:00
parent de19ef2931
commit 8ba60d3caa
5 changed files with 16 additions and 0 deletions

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 19 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

View File

@ -1991,6 +1991,10 @@ class UnsortedSegmentSum(PrimitiveWithInfer):
up. Segment_ids does not need to be sorted, and it does not need to cover all values in the entire valid value
range.
The following figure shows the calculation process of UnsortedSegmentSum:
.. image:: api_img/UnsortedSegmentSum.png
Note:
If the segment_id i is absent in the segment_ids, then output[i] will be filled with 0.
@ -2091,6 +2095,10 @@ class UnsortedSegmentMin(PrimitiveWithCheck):
"""
Computes the minimum of a tensor along segments.
The following figure shows the calculation process of UnsortedSegmentMin:
.. image:: api_img/UnsortedSegmentMin.png
Note:
If the segment_id i is absent in the segment_ids, then output[i] will be filled with
the maximum value of the input_x's type.
@ -2150,6 +2158,10 @@ class UnsortedSegmentMax(PrimitiveWithCheck):
"""
Computes the maximum along segments of a tensor.
The following figure shows the calculation process of UnsortedSegmentMax:
.. image:: api_img/UnsortedSegmentMax.png
Note:
If the segment_id i is absent in the segment_ids, then output[i] will be filled with
the minimum value of the input_x's type.
@ -2210,6 +2222,10 @@ class UnsortedSegmentProd(PrimitiveWithInfer):
"""
Computes the product of a tensor along segments.
The following figure shows the calculation process of UnsortedSegmentProd:
.. image:: api_img/UnsortedSegmentProd.png
Inputs:
- **input_x** (Tensor) - The shape is :math:`(x_1, x_2, ..., x_R)`.
With float16, float32 or int32 data type.