forked from mindspore-Ecosystem/mindspore
add the picture of calculation for UnsortedSegmentxxx.
This commit is contained in:
parent
de19ef2931
commit
8ba60d3caa
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 |
|
@ -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.
|
||||
|
|
Loading…
Reference in New Issue