ShapeOf

バージョン名: ShapeOf-1

カテゴリー: 形状の操作

詳細な説明: ShapeOf は、入力テンソル形状を使用して 1D テンソルを生成します。

属性: 属性がありません。

入力:

  • 1: 任意の入力テンソル。必須。

出力:

  • 1: 入力テンソル形状と等しい 1D テンソル。要素の数は入力テンソルランクと同じです。入力テンソルがスカラー、つまり 0 次元テンソルの場合は、空の 1D テンソルにすることができます。

例:

 <layer ... type="ShapeOf">
     <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>