NonZero

バージョン名: NonZero-3

カテゴリー: 条件

簡単な説明: NonZero は、入力テンソルの非ゼロ要素のインデックスを返します。

詳細な説明: NonZero は、入力テンソルの非ゼロ要素のインデックスを (行優先の順序で、次元ごとに) 返します。

  • 出力テンソルの形状は [rank(input), num_non_zero] です。

  • 例えば、テンソル [[1, 0], [1, 1]] の場合、出力は [[0, 1, 1], [0, 0, 1]] になります。

  • 出力はタプルのコレクションであり、各タプルには rank(input) 要素があり、単一の非ゼロ要素のインデックスが含まれます。

  • 各出力次元の i 番目の要素は、i 番目のタプルの一部です。

  • 与えられた例では、タプルは [0, 0][1, 0][1, 1] になります。

属性:

  • output_type

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

    • 値の範囲: i64 または i32

    • タイプ: 文字列

    • デフォルト値: “i64”

    • 必須: いいえ

入力:

  • 1: タイプ T の任意の形状のテンソル。必須。

出力:

  • 1: T_OUT タイプの形状 [rank(data), num_non_zero] の非ゼロ要素のインデックスを持つテンソル。

タイプ:

  • T: 任意のタイプ。

  • T_OUT: output_type 属性に応じて、int64 または int32 を指定できます。

例:

 <layer ... type="NonZero">
     <data output_type="i64"/>
     <input>
         <port id="0">
             <dim>3</dim>
             <dim>10</dim>
             <dim>100</dim>
             <dim>200</dim>
         </port>
     </input>
     <output>
         <port id="1">
             <dim>4</dim>
             <dim>-1</dim>
         </port>
     </output>
 </layer>