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>