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 or int32.

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>