パブリック最終クラスTopKUUnique
配列内の TopK の一意の値を並べ替えられた順序で返します。
実行時間は、K と入力サイズの積に比例します。 K の値が十分に大きい場合は、配列全体をソートする方が効率的です。中央値対中央値アルゴリズムの方がおそらく高速ですが、XLA で効率的に実装するのは困難です。一意の数値 (NAN ではない) が K 個よりも少ない場合、結果は負の無限大で埋められます。 NaN は決して返されません。非正規数はゼロにフラッシュされます。要素が複数のインデックスに出現する場合は、最大のインデックスが返されます。パディング値により TopK 要素が入力に表示されない場合、インデックスは負の 1 でパディングされます。パディング値が入力に現れ、パディングが必要な場合は、パディング値の最大のインデックスが返されます。セマンティクスは kth_order_statistic と同じではありません。
パブリックメソッド
静的TopKUnique | |
出力<浮動小数点> | トップク() |
出力<整数> | topkIndices () |
継承されたメソッド
パブリックメソッド
public static TopKUnique create (スコープscope、オペランド<Float>入力、Long k)
新しい TopKUnique オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|
戻り値
- TopKUnique の新しいインスタンス