W tym blogu przedstawimy przykładową metodę użycia metody „torch.argmax()” w PyTorch.
Jak korzystać z metody „torch.argmax()” w PyTorch?
Metoda „torch.argmax()” przyjmuje jako dane wejściowe dowolny tensor 1D lub 2D i zwraca tensor zawierający indeksy/indeksy wartości maksymalnych wzdłuż danego wymiaru.
Poniżej podana jest składnia metody „torch.argmax()”:
latarka. argmax ( < wejściowy_tensor > )
Aby użyć tej metody w PyTorch, przejrzyj poniższe przykłady dla lepszego zrozumienia:
Przykład 1: Użyj metody „torch.argmax()” z tensorem 1D
W pierwszym przykładzie utworzymy tensor 1D i użyjemy z nim metody „torch.argmax()”. Postępujmy zgodnie z poniższą procedurą krok po kroku:
Krok 1: Zaimportuj bibliotekę PyTorch
Najpierw zaimportuj „ latarka ”, aby użyć metody „torch.argmax()”:
import latarkaKrok 2: Utwórz tensor 1D
Następnie utwórz tensor 1D i wydrukuj jego elementy. Tutaj tworzymy następujące „ Dziesiątki1 ” tensor z listy przy użyciu „ tensor palnika() funkcja:
Dziesiątki1 = latarka. napinacz ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )
wydrukować ( Dziesiątki1 )
Stworzyło to tensor 1D, jak widać poniżej:
Krok 3: Znajdź indeksy o maksymalnej wartości
Teraz skorzystaj z opcji „ latarka.argmax() ”, aby znaleźć indeks/indeksy wartości maksymalnej w „ Dziesiątki1 tensor:
T1_ind = latarka. argmax ( Dziesiątki1 )Krok 4: Wydrukuj indeks o maksymalnej wartości
Na koniec wyświetl indeks wartości maksymalnej w tensorze wejściowym:
wydrukować ( „Indeksy:” , T1_ind )Poniższe dane wyjściowe pokazują indeks wartości maksymalnej w „ Dziesiątki1 ” tensor, tj. 4. Oznacza to, że najwyższa wartość tensora znajduje się przy 4. indeksie, czyli „ 9 ”:
Przykład 2: Użyj metody „torch.argmax()” z tensorem 2D
W drugim przykładzie utworzymy tensor 2D i użyjemy z nim metody „torch.argmax()”. Postępujmy zgodnie z podanymi krokami:
Krok 1: Zaimportuj bibliotekę PyTorch
Najpierw zaimportuj „ latarka ”, aby użyć metody „torch.argmax()”:
import latarkaKrok 2: Utwórz tensor 2D
Następnie skorzystaj z opcji „ tensor palnika() ”, aby utworzyć tensor 2D i wydrukować jego elementy. Tutaj tworzymy następujące „ Tens2 „Tensor 2D:
Tens2 = latarka. napinacz ( [ [ 4 , 1 , - 7 ] , [ piętnaście , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )wydrukować ( Tens2 )
Stworzyło to tensor 2D, jak widać poniżej:
Krok 3: Znajdź indeksy o maksymalnej wartości
Teraz znajdź indeks maksymalnej wartości w „ Tens2 ” tensor, wykorzystując „ latarka.argmax() funkcja:
T2_ind = latarka. argmax ( Tens2 )Krok 4: Wydrukuj indeks o maksymalnej wartości
Na koniec wyświetl indeks wartości maksymalnej w tensorze wejściowym:
wydrukować ( „Indeksy:” , T2_ind )Zgodnie z poniższym wyjściem, indeks wartości maksymalnej w „ Tens2 tensor wynosi „3”. Oznacza to, że najwyższa wartość tensora występuje przy trzecim indeksie, czyli „ piętnaście ”:
Krok 5: Znajdź indeksy o maksymalnej wartości wzdłuż kolumn
Co więcej, użytkownicy mogą również znaleźć indeksy/indeksy wartości maksymalnych wzdłuż każdej kolumny tensora. Na przykład możemy użyć „ przyćmione=0 ” z funkcją „torch.argmax()”. Znajduje indeksy wartości maksymalnych wzdłuż kolumn w „ Tens2 ” tensor, a następnie wypisuje te indeksy:
indeks_kol = latarka. argmax ( Tens2 , ciemny = 0 )wydrukować ( „Indeksy w kolumnach:” , indeks_kol )
Poniższe dane wyjściowe pokazują indeksy wartości maksymalnych wzdłuż każdej kolumny tensora:
Krok 6: Znajdź indeksy o maksymalnej wartości wzdłuż wierszy
Podobnie użytkownicy mogą również znaleźć indeksy/indeksy wartości maksymalnych wzdłuż każdego wiersza tensora. Na przykład użyj opcji „ przyćmione=1 ” za pomocą funkcji „torch.argmax()”, aby znaleźć indeksy wartości maksymalnych wzdłuż wierszy tensora „Tens2”, a następnie wydrukować te indeksy:
indeks_wiersza = latarka. argmax ( Tens2 , ciemny = 1 )wydrukować ( „Indeksy w wierszach:” , indeks_wiersza )
Indeksy wartości maksymalnej wzdłuż każdego wiersza tensora „Tens2” można zobaczyć poniżej:
Sprawnie wyjaśniliśmy metodę użycia metody „torch.argmax()” w PyTorch.
Notatka : tutaj możesz uzyskać dostęp do naszego Notatnika Google Colab połączyć .
Wniosek
Aby użyć metody „torch.argmax()” w PyTorch, najpierw zaimportuj „ latarka ' biblioteka. Następnie utwórz żądany tensor 1D lub 2D i przejrzyj jego elementy. Następnie użyj opcji „ latarka.argmax() ” metoda znajdowania/obliczania indeksów/indeksów wartości maksymalnych w tensorze. Co więcej, użytkownicy mogą również znaleźć indeksy wartości maksymalnej wzdłuż każdego wiersza lub kolumny tensora, korzystając z „ ciemny ' argument. Na koniec wyświetl indeks wartości maksymalnej w tensorze wejściowym. W tym blogu przedstawiono przykładową metodę użycia metody „torch.argmax()” w PyTorch.