torch.topkとtorch.quantileの違い

torch.topk — PyTorch 2.9 documentation
torch.quantile — PyTorch 2.9 documentation

Question

torch.topk と torch.quantile の違いについてまとめて

特徴	torch.topk	torch.quantile
目的	テンソルの特定次元で、値が大きい（または小さい）順の上位 $k$ 個の要素を取得する。	テンソルの特定次元で、指定された**パーセンタイル（分位点）**に対応する値を取得する。
入力パラメータ	取得したい要素の数 $k$ を指定する。	取得したい分位点（0.0～1.0）または分位点のリスト $q$ を指定する。
返す情報	上位 $k$ 個の要素の値と、それらの元のインデックスを返す。	指定された分位点 $q$ の値を返す（インデックスは返さない）。
使用例	最大の要素や、上位 10% の要素などを直接知りたい場合。	* データの中央値（0.5 分位点）を知りたい場合。* 外れ値の境界（例えば、95 パーセンタイル）を特定したい場合。