ShapeOf#

バージョン名: ShapeOf-3

カテゴリー: 形状操作

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

属性:

  • output_type

    • 説明: 出力テンソルのタイプ

    • 値の範囲: “i64” or “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"> <!-- 出力値: [2,3,224,224]--> 
            <dim>4</dim> 
        </port> 
    </output> 
</layer>