Revise Unsqueeze op - specification (#5526)
* Revise unsqueeze op spec * Second input value boundaries fix * adjust first input description
This commit is contained in:
parent
84b94c9834
commit
af0aa5f430
@ -10,9 +10,9 @@
|
||||
|
||||
**Inputs**:
|
||||
|
||||
* **1**: Multidimensional input tensor of type *T*. *Required*.
|
||||
* **1**: Tensor of type *T* and arbitrary shape. **Required**.
|
||||
|
||||
* **2**: OD or 1D tensor of type *T_SHAPE* with dimensions indices to be set to 1. Values could be negative. *Required*.
|
||||
* **2**: Scalar or 1D tensor of type *T_INT* with indices of dimensions to unsqueeze. Values could be negative (have to be from range `[-R, R-1]`, where `R` is the rank of the output). **Required**.
|
||||
|
||||
**Outputs**:
|
||||
|
||||
@ -20,13 +20,13 @@
|
||||
|
||||
**Types**
|
||||
|
||||
* *T*: supported type.
|
||||
* *T*: any numeric type.
|
||||
|
||||
* *T_SHAPE*: supported integer type.
|
||||
* *T_INT*: any supported integer type.
|
||||
|
||||
**Example**
|
||||
|
||||
*Example 1:*
|
||||
*Example 1: unsqueeze 2D tensor to a 4D tensor*
|
||||
```xml
|
||||
<layer ... type="Unsqueeze">
|
||||
<input>
|
||||
@ -51,7 +51,7 @@
|
||||
</layer>
|
||||
```
|
||||
|
||||
*Example 2: (unsqueeze 0D tensor (constant) to 1D tensor)*
|
||||
*Example 2: unsqueeze 0D tensor (constant) to 1D tensor*
|
||||
```xml
|
||||
<layer ... type="Unsqueeze">
|
||||
<input>
|
||||
|
Loading…
Reference in New Issue
Block a user