トップKUユニーク

パブリック最終クラスTopKUUnique

配列内の TopK の一意の値を並べ替えられた順序で返します。の

実行時間は、K と入力サイズの積に比例します。 K の値が十分に大きい場合、配列全体をソートする方が効率的です。中央値対中央値アルゴリズムの方がおそらく高速ですが、XLA で効率的に実装するのは困難です。一意の数値 (NAN ではない) が K 個よりも少ない場合、結果は負の無限大で埋められます。 NaN は決して返されません。非正規数はゼロにフラッシュされます。要素が複数のインデックスに出現する場合は、最大のインデックスが返されます。パディング値により TopK 要素が入力に表示されない場合、インデックスは負の 1 でパディングされます。パディング値が入力に現れ、パディングが必要な場合は、パディング値の最大のインデックスが返されます。セマンティクスは kth_order_statistic と同じではありません。

パブリックメソッド

静的TopKUnique
create (スコープscope、オペランド<Float>入力、Long k)
新しい TopKUnique オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<浮動小数点>
出力<整数>

継承されたメソッド

パブリックメソッド

public static TopKUnique create (スコープscope、オペランド<Float>入力、Long k)

新しい TopKUnique オペレーションをラップするクラスを作成するためのファクトリ メソッド。

パラメーター
範囲現在のスコープ
戻り値
  • TopKUnique の新しいインスタンス

public出力<Float> topk ()

public Output <整数> topkIndices ()