Parameter

バージョン名: Parameter-1

カテゴリー: インフラストラクチャー

簡単な説明: Parameter レイヤーはモデルの入力を指定します。

属性:

• element_type

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

  • 値の範囲: u1、u4、u8、u16、u32、u64、i4、i8、i16、i32、i64、f16、f32、boolean、bf16

  • タイプ: string

  • 必須: はい

• shape

  • 説明: 出力テンソルの形状

  • 値の範囲: 負ではない整数のリスト、空のリストが許可されます。これは 0D またはスカラーテンソルを意味します

  • タイプ: int[]

  • 必須: はい

出力:

• 1: タイプ T および shape 属性と等しい形状の出力テンソル。

タイプ:

• T: 要素タイプ値からの任意のタイプ。

例:

<layer ... type="Parameter" ...>
    <data>element_type="f32" shape="1,3,224,224"</data>
    <output>
        <port id="0">
            <dim>1</dim>
            <dim>3</dim>
            <dim>224</dim>
            <dim>224</dim>
        </port>
    </output>
</layer>