ShapeOf

Versioned name: ShapeOf-3

Category: Shape manipulation

Short description: ShapeOf produces 1D tensor with the input tensor shape.

Attributes:

Inputs:

Outputs:

Types

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">
<dim>4</dim>
</port>
</output>
</layer>