Kaip „PyTorch“ naudoti „torch.argmax()“ metodą?

Šiame tinklaraštyje bus parodytas „torch.argmax()“ metodo „PyTorch“ naudojimo pavyzdys.

Kaip „PyTorch“ naudoti „torch.argmax()“ metodą?

Metodas „torch.argmax()“ priima bet kurį 1D arba 2D tenzorių kaip įvestį ir grąžina tenzorį, kuriame yra didžiausių reikšmių pagal nurodytą matmenį indeksai / indeksai.

Metodo „torch.argmax()“ sintaksė pateikta toliau:

Norėdami naudoti šį metodą PyTorch, peržiūrėkite šiuos pavyzdžius, kad geriau suprastumėte:

1 pavyzdys: naudokite „torch.argmax()“ metodą su 1D tensor

Pirmajame pavyzdyje sukursime 1D tenzorių ir su juo naudosime „torch.argmax()“ metodą. Atlikime toliau pateiktą žingsnis po žingsnio procedūrą:

1 veiksmas: importuokite PyTorch biblioteką

Pirmiausia importuokite „ deglas “ biblioteką, kad galėtumėte naudoti „torch.argmax()“ metodą:

2 veiksmas: sukurkite 1D tenzorį

Tada sukurkite 1D tenzorių ir atspausdinkite jo elementus. Čia mes kuriame šiuos ' Dešimtys 1 “ tenzorius iš sąrašo naudojant “ torch.tensor() ' funkcija:

Tai sukūrė 1D tenzorių, kaip parodyta toliau:

3 veiksmas: suraskite didžiausios vertės indeksus

Dabar naudokite ' torch.argmax() ' funkcija, kad surastumėte didžiausios vertės indeksą / indeksus ' Dešimtys 1 “ tenzorius:

4 veiksmas: spausdinkite didžiausios vertės indeksą

Galiausiai įvesties tensoryje parodykite didžiausios vertės indeksą:

Žemiau pateikta išvestis rodo didžiausios vertės indeksą „ Dešimtys 1 “ tenzorius, ty 4. Tai reiškia, kad didžiausia tenzoriaus reikšmė yra ties 4 indeksu, kuris yra “ 9 “:

2 pavyzdys: naudokite „torch.argmax()“ metodą su 2D tensor

Antrame pavyzdyje sukursime 2D tenzorių ir su juo naudosime „torch.argmax()“ metodą. Atlikime nurodytus veiksmus:

1 veiksmas: importuokite PyTorch biblioteką

Pirmiausia importuokite „ deglas “ biblioteką, kad galėtumėte naudoti „torch.argmax()“ metodą:

2 veiksmas: sukurkite 2D tenzorį

Tada naudokite „ torch.tensor() “ funkcija sukurti 2D tenzorių ir spausdinti jo elementus. Čia mes kuriame šiuos ' Dešimtys2 „2D tenzorius:

Tai sukūrė 2D tenzorių, kaip parodyta toliau:

3 veiksmas: suraskite didžiausios vertės indeksus

Dabar raskite didžiausios vertės indeksą „ Dešimtys2 “ tenzorius naudojant “ torch.argmax() ' funkcija:

4 veiksmas: spausdinkite didžiausios vertės indeksą

Galiausiai įvesties tensoryje parodykite didžiausios vertės indeksą:

Pagal toliau pateiktą išvestį didžiausios vertės indeksas „ Dešimtys2 “ tenzorius yra „3“. Tai reiškia, kad didžiausia tenzoriaus reikšmė yra ties 3 indeksu, kuris yra ' penkiolika “:

5 veiksmas: raskite didžiausios vertės indeksus stulpeliuose

Be to, vartotojai taip pat gali rasti didžiausių verčių indeksus / indeksus kiekviename tenzoriaus stulpelyje. Pavyzdžiui, galime naudoti „ dim=0 ” argumentą su funkcija „torch.argmax()“. Jis randa didžiausių reikšmių indeksus stulpeliuose „ Dešimtys2 “ tenzorius ir išspausdina šiuos indeksus:

Toliau pateiktame išvestyje rodomi didžiausių verčių indeksai kiekviename tenzoriaus stulpelyje:

6 veiksmas: raskite didžiausios vertės indeksus išilgai eilučių

Panašiai vartotojai taip pat gali rasti didžiausių reikšmių indeksus / indeksus kiekvienoje tenzoriaus eilutėje. Pavyzdžiui, naudokite „ dim=1 “ argumentą su funkcija „torch.argmax()“, kad surastumėte didžiausių reikšmių indeksus tensoriaus „Tens2“ eilutėse ir tada išspausdintumėte šiuos indeksus:

Toliau pateikiami didžiausios vertės indeksai kiekvienoje „Tens2“ tenzoriaus eilutėje:

Mes efektyviai paaiškinome metodą, kaip naudoti „torch.argmax()“ metodą „PyTorch“.

Pastaba : „Google Colab“ bloknotą galite pasiekti šiuo adresu nuoroda .

Išvada

Norėdami naudoti „torch.argmax()“ metodą „PyTorch“, pirmiausia importuokite „ deglas “ biblioteka. Tada sukurkite norimą 1D arba 2D tenzorių ir peržiūrėkite jo elementus. Tada naudokite „ torch.argmax() “ metodas, leidžiantis rasti / apskaičiuoti didžiausių tenzorio reikšmių indeksus / indeksus. Be to, vartotojai taip pat gali rasti didžiausios vertės indeksus kiekvienoje tenzoriaus eilutėje ar stulpelyje naudodami „ pritemdyta “ argumentas. Galiausiai įvesties tensoryje parodykite didžiausios vertės indeksą. Šiame tinklaraštyje parodytas „torch.argmax()“ metodo „PyTorch“ naudojimo pavyzdys.