ShapeOf#
Versioned name: ShapeOf-3
Category: Shape manipulation
Short description: ShapeOf produces 1D tensor with the input tensor shape.
Attributes:
output_type
Description: the output tensor type
Range of values: “i64” or “i32”
Type: string
Default value: “i64”
Required: no
Inputs:
1: Arbitrary input tensor of type T. Required.
Outputs:
1: 1D tensor that is equal to input tensor shape of type T_IND. Number of elements is equal to input tensor rank. Can be empty 1D tensor if input tensor is a scalar, that means 0-dimensional tensor.
Types
T: any numeric type.
T_IND:
int64
orint32
.
Example
<layer ... type="ShapeOf">
<data output_type="i64"/>
<input>
<port id="0">
<dim>2</dim>
<dim>3</dim>
<dim>224</dim>
<dim>224</dim>
</port>
</input>
<output>
<port id="1"> <!-- output value is: [2,3,224,224]-->
<dim>4</dim>
</port>
</output>
</layer>