update Operators.md

This commit is contained in:
Tang 2022-08-01 15:06:42 +08:00 committed by Sven
parent ecfc8735d9
commit 128d19b448
1 changed files with 86 additions and 0 deletions

View File

@ -27,6 +27,7 @@
- [Erf](#erf)
- [FullyConnected](#fullyconnected)
- [Gather](#gather)
- [GatherElements](#gatherelements)
- [GatherNd](#gathernd)
- [GroupedConv1d](#groupedconv1d)
- [GroupedConv2d](#groupedconv2d)
@ -36,7 +37,9 @@
- [Or](#or)
- [LogSoftmax](#logsoftmax)
- [Matmul](#matmul)
- [MaxpooGrad](#maxpoograd)
- [MaxpoolWithArgmax](#maxpoolwithargmax)
- [MaxpoolWithArgmax2](#maxpoolwithargmax2)
- [MaxUnpool2d](#maxunpool2d)
- [Moments](#moments)
- [NBG](#nbg)
@ -64,6 +67,8 @@
- [Resize](#resize)
- [Resize1d](#resize1d)
- [Reverse](#reverse)
- [RoiAlign](#roialign)
- [RoiPool](#roipool)
- [ScatterND](#scatternd)
- [Select](#select)
- [DataConvert](#dataconvert)
@ -77,6 +82,7 @@
- [Square](#square)
- [LogicalNot](#logicalnot)
- [Floor](#floor)
- [Ceil](#ceil)
- [Cast](#cast)
- [Slice](#slice)
- [Softmax](#softmax)
@ -88,6 +94,7 @@
- [StridedSlice](#stridedslice)
- [Svdf](#svdf)
- [Tile](#tile)
- [Topk](#topk)
- [Transpose](#transpose)
- [Unidirectional sequence lstm](#unidirectional-sequence-lstm)
- [Unstack](#unstack)
@ -131,6 +138,10 @@ Prelu(x) : alpha * x if x <= 0; x if x > 0. alpha is a tensor.
Linear(x, a, b) : a*x + b.
Gelu(x) : x * P(X <= x), where P(x) ~ N(0, 1). https://tensorflow.google.cn/api_docs/python/tf/nn/gelu
Selu(x, alpha, gamma) : gamma * x if(x>=0), gamma * alpha * (exp(x)-1) x<0
Celu(x, alpha) : x if x >= 0; alpha * (exp(x/alpha) - 1)
```
<a class="mk-toclify" id="addn"></a>
@ -359,6 +370,15 @@ input tensor with each element in the output tensor.
Gather slices from input, **axis** according to **indices**.
<a class="mk-toclify" id="gatherelements"></a>
## GatherElements
GatherElements slices from input, **axis** according to **indices**.
out[i][j][k] = input[index[i][j][k]][j][k] if axis = 0,
out[i][j][k] = input[i][index[i][j][k]][k] if axis = 1,
out[i][j][k] = input[i][j][index[i][j][k]] if axis = 2,
https://github.com/onnx/onnx/blob/main/docs/Operators.md#GatherElements
<a class="mk-toclify" id="gathernd"></a>
## GatherNd
@ -455,6 +475,22 @@ Multiplies matrix a by matrix b, producing a * b.
- adjoint_a: If True, a is conjugated and transposed before multiplication.
- adjoint_b: If True, b is conjugated and transposed before multiplication.
<a class="mk-toclify" id="maxpoograd"></a>
## MaxpooGrad
Acquire the gradient of 2-D Max pooling operation's input tensor. \
Like the tensorflow_XLA op SelectAndScatter, see https://tensorflow.google.cn/xla/operation_semantics?hl=en#selectandscatter.
- padding : AUTO, VALID or SAME.
- ksize : filter size.
- stride : stride along each spatial axis.
- round_type : CEILING or FLOOR.
* Inputs:
- 0 : input tensor of 2-D Max pooling.
- 1 : gradient of 2-D Max pooling output tensor.
<a class="mk-toclify" id="maxpoolwithargmax"></a>
## MaxpoolWithArgmax
@ -465,6 +501,16 @@ Performs an 2-D Max pooling operation and return indices
- stride : stride along each spatial axis.
- round_type : CEILING or FLOOR.
<a class="mk-toclify" id="maxpoolwithargmax2"></a>
## MaxpoolWithArgmax2
Performs an 2-D Max pooling operation and return indices(which start at the beginning of the input tensor).
- padding : AUTO, VALID or SAME.
- ksize : filter size.
- stride : stride along each spatial axis.
- round_type : CEILING or FLOOR.
<a class="mk-toclify" id="maxunpool2d"></a>
## MaxUnpool2d
@ -688,6 +734,34 @@ Reverses specific dimensions of a tensor.
- axis : The indices of the dimensions to reverse.
<a class="mk-toclify" id="roialign"></a>
## RoiAlign
Select and scale the feature map of each region of interest to a unified output
size by average pooling sampling points from bilinear interpolation.
- output_height : specifying the output height of the output tensor.
- output_width : specifying the output width of the output tensor.
- height_ratio : specifying the ratio from the height of original image to the
height of feature map.
- width_ratio : specifying the ratio from the width of original image to the
width of feature map.
- height_sample_num : specifying the number of sampling points in height dimension
used to compute the output.
- width_sample_num :specifying the number of sampling points in width dimension
used to compute the output.
<a class="mk-toclify" id="roipool"></a>
## RoiPool
Select and scale the feature map of each region of interest to a unified output
size by max-pooling.
pool_type : only support max-pooling (MAX)
scale : The ratio of image to feature map (Range: 0 < scale <= 1)
size : The size of roi pooling (height/width)
<a class="mk-toclify" id="scatternd"></a>
## ScatterND
@ -756,6 +830,11 @@ LogicalNot(x) : NOT x
returns the largest integer less than or equal to a given number.
<a class="mk-toclify" id="ceil"></a>
## Ceil
returns the largest integer more than or equal to a given number.
<a class="mk-toclify" id="cast"></a>
## Cast
@ -863,6 +942,13 @@ Constructs a tensor by tiling a given tensor.
- multiples : Must be one of the following types: int32, int64.
Length must be the same as the number of dimensions in input.
<a class="mk-toclify" id="topk"></a>
## Topk
Finds values and indices of the k largest entries for the last dimension.
- k : Number of top elements to look for along the last dimension.
<a class="mk-toclify" id="transpose"></a>
## Transpose