TopKUnique

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

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