py argsort() 関数の使用法

この関数は配列に作用します。配列は np データまたはトーチ データである場合もあります

関数の後、同じ次元のデータが取得され、各位置の要素は元の配列の行の要素のサイズのオーダーになります。

この関数をよりよく説明するために、0 から n-1 で構成される n 次元のテンソル データを生成できる torch.randperm(n) 関数を使用します。

たとえば、取得した x の最大値は (0 から数えて) 最初の位置にあり、次に 1 が x.argsort() の最後の位置にあります。
x の最小値は 3 番目の位置にあるため、3 は x.argsort() の 0 番目の位置にあります。

入力は行列であり、出力は同じです。

スタック関数は、指定された次元に従ってテンソルをスタックできます。

いいねは作者だけでなく読者にとっても意味のあるものです。
もし過去に戻れるなら、私を導いてくれたあの記事をぜひ読みたいです