ShapeOf¶
バージョン名: ShapeOf-3
カテゴリー: 形状の操作
詳細な説明: ShapeOf は、入力テンソル形状を使用して 1D テンソルを生成します。
属性:
-
output_type
説明: 出力テンソルのタイプ
値の範囲: “i64” または “i32”
タイプ: 文字列
デフォルト値: “i64”
必須: いいえ
入力:
1: T タイプの任意の入力テンソル。必須。
出力:
1: T_IND タイプの入力テンソル形状と等しい 1D テンソル。要素の数は入力テンソルランクと同じです。入力テンソルがスカラー、つまり 0 次元テンソルの場合は、空の 1D テンソルにすることができます。
タイプ:
T: 任意の数値タイプ。
T_IND:
int64
またはint32
。
例:
<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>