このブログでは、PyTorchの「torch.argmax()」メソッドを使用する方法を例に説明します。
PyTorch で「torch.argmax()」メソッドを使用するには?
「torch.argmax()」メソッドは、任意の 1D または 2D テンソルを入力として受け取り、指定された次元に沿った最大値のインデックスを含むテンソルを返します。
「torch.argmax()」メソッドの構文は以下のとおりです。
松明。 引数最大 ( < 入力テンソル > )
PyTorch でこのメソッドを使用するには、理解を深めるために次の例を参照してください。
例 1: 1D テンソルで「torch.argmax()」メソッドを使用する
最初の例では、1D テンソルを作成し、それとともに「torch.argmax()」メソッドを使用します。以下の段階的な手順に従ってください。
ステップ 1: PyTorch ライブラリをインポートする
まず、「」をインポートします 松明 「torch.argmax()」メソッドを使用するには、「ライブラリ」を使用します。
輸入 松明ステップ 2: 1D テンソルを作成する
次に、1D テンソルを作成し、その要素を出力します。ここでは以下のようなものを作成しています。 十の位1 ” を使用してリストからテンソルを取得します。 torch.tensor() ' 関数：
十の位1 = 松明。 テンソル ( [ 5 、 0 、 - 8 、 1 、 9 、 7 ] )
印刷する ( 十の位1 )
これにより、以下に示すように 1D テンソルが作成されました。
ステップ 3: 最大値のインデックスを見つける
さあ、「」を活用してください。 トーチ.argmax() 」関数で最大値のインデックスを見つけます。 十の位1 ” テンソル:
T1_ind = 松明。 引数最大 ( 十の位1 )ステップ 4: 最大値のインデックスを出力する
最後に、入力テンソルの最大値のインデックスを表示します。
印刷する ( 「インデックス:」 、 T1_ind )以下の出力は、「」の最大値のインデックスを示しています。 十の位1 ” テンソル、つまり 4。これは、テンソルの最高値が 4 番目のインデックスにあることを意味します。 9 ”:
例 2: 2D テンソルで「torch.argmax()」メソッドを使用する
2 番目の例では、2D テンソルを作成し、それとともに「torch.argmax()」メソッドを使用します。提供された手順に従いましょう。
ステップ 1: PyTorch ライブラリをインポートする
まず、「」をインポートします 松明 「torch.argmax()」メソッドを使用するには、「ライブラリ」を使用します。
輸入 松明ステップ 2: 2D テンソルを作成する
次に、「」を使用します。 torch.tensor() 」関数を使用して 2D テンソルを作成し、その要素を出力します。ここでは以下のようなものを作成しています。 十数2 「2D テンソル:
十数2 = 松明。 テンソル ( [ [ 4 、 1 、 - 7 ] 、 [ 15 、 6 、 0 ] 、 [ - 7 、 9 、 2 ] ] )印刷する ( 十数2 )
これにより、以下に示すような 2D テンソルが作成されました。
ステップ 3: 最大値のインデックスを見つける
ここで、「」の最大値のインデックスを見つけます。 十数2 「」を利用した「テンソル」 トーチ.argmax() ' 関数：
T2_ind = 松明。 引数最大 ( 十数2 )ステップ 4: 最大値のインデックスを出力する
最後に、入力テンソルの最大値のインデックスを表示します。
印刷する ( 「インデックス:」 、 T2_ind )以下の出力によると、「」の最大値のインデックスは 十数2 ” テンソルは「3」です。これは、テンソルの最高値が 3 番目のインデックスにあることを意味します。 15 ”:
ステップ 5: 列に沿った最大値のインデックスを見つける
さらに、ユーザーはテンソルの各列に沿った最大値のインデックスを見つけることもできます。たとえば、「」を使用できます。 ディム=0 」引数を「torch.argmax()」関数で指定します。 「」の列に沿って最大値のインデックスを見つけます。 十数2 ” tensor を計算し、それらのインデックスを出力します。
列インデックス = 松明。 引数最大 ( 十数2 、 薄暗い = 0 )印刷する ( 「列のインデックス:」 、 列インデックス )
以下の出力は、テンソルの各列に沿った最大値のインデックスを示しています。
ステップ 6: 行に沿って最大値のインデックスを見つける
同様に、ユーザーはテンソルの各行に沿った最大値のインデックスを見つけることもできます。たとえば、「」を使用します。 ディム=1 」引数を「torch.argmax()」関数で使用して、「Tens2」テンソルの行に沿って最大値のインデックスを見つけて、それらのインデックスを出力します。
行インデックス = 松明。 引数最大 ( 十数2 、 薄暗い = 1 )印刷する ( 「行のインデックス:」 、 行インデックス )
「Tens2」テンソルの各行に沿った最大値のインデックスを以下に示します。
PyTorchの「torch.argmax()」メソッドを利用する方法を効率よく解説しました。
注記 : ここから Google Colab ノートブックにアクセスできます リンク 。
結論
PyTorch で「torch.argmax()」メソッドを使用するには、まず「 松明 ' 図書館。次に、目的の 1D または 2D テンソルを作成し、その要素を表示します。次に、「」を使用します。 トーチ.argmax() 」メソッドを使用して、テンソルの最大値のインデックスを検索/計算します。さらに、ユーザーは「」を使用してテンソルの各行または列に沿った最大値のインデックスを見つけることもできます。 薄暗い ' 口論。最後に、入力テンソルの最大値のインデックスを表示します。このブログでは、PyTorchの「torch.argmax()」メソッドを使用する方法を例示しました。