Ez a blog példát mutat be a PyTorch „torch.argmax()” metódusának használatára.
Hogyan kell használni a „torch.argmax()” módszert a PyTorch-ban?
A „torch.argmax()” metódus bármely 1D vagy 2D tenzort vesz bemenetként, és egy olyan tenzort ad vissza, amely tartalmazza az adott dimenzió mentén a maximális értékek indexeit/indexeit.
A „torch.argmax()” metódus szintaxisa az alábbiakban látható:
fáklya. argmax ( < bemeneti_tenzor > )
A módszer PyTorch alkalmazásban való használatához a jobb megértés érdekében nézze át a következő példákat:
1. példa: Használja a „torch.argmax()” módszert az 1D tenzorral
Az első példában létrehozunk egy 1D tenzort, és a „torch.argmax()” metódust használjuk hozzá. Kövessük az alábbi lépésenkénti eljárást:
1. lépés: Importálja a PyTorch könyvtárat
Először importálja a „ fáklya ” könyvtár a „torch.argmax()” metódus használatához:
import fáklya2. lépés: Hozzon létre 1D tenzort
Ezután hozzon létre egy 1D tenzort, és nyomtassa ki az elemeit. Itt a következőt hozzuk létre: Tens1 ' tenzor egy listából a ' torch.tensor() ” funkció:
Tens1 = fáklya. tenzor ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )
nyomtatás ( Tens1 )
Ezzel létrejött egy 1D tenzor, amint az alább látható:
3. lépés: Keresse meg a maximális értékű indexeket
Most használja a „ torch.argmax() ' függvény segítségével megkeresheti a maximális érték indexét/indexeit a ' Tens1 ” tenzor:
T1_ind = fáklya. argmax ( Tens1 )4. lépés: Nyomtassa ki a maximális érték indexét
Végül jelenítse meg a maximális érték indexét a bemeneti tenzorban:
nyomtatás ( 'Indexek:' , T1_ind )Az alábbi kimenet a maximális érték indexét mutatja a „ Tens1 ” tenzor, azaz 4. Ez azt jelenti, hogy a tenzor legmagasabb értéke a 4. indexnél van, amely „ 9 ”:
2. példa: Használja a „torch.argmax()” módszert a 2D Tensorral
A második példában létrehozunk egy 2D tenzort, és a „torch.argmax()” metódust használjuk vele. Kövessük a megadott lépéseket:
1. lépés: Importálja a PyTorch könyvtárat
Először importálja a „ fáklya ” könyvtár a „torch.argmax()” metódus használatához:
import fáklya2. lépés: Hozzon létre 2D tenzort
Ezután használja a „ torch.tensor() ” funkció segítségével 2D tenzort hozhat létre és kinyomtathatja annak elemeit. Itt a következőt hozzuk létre: Tens2 „2D tenzor:
Tens2 = fáklya. tenzor ( [ [ 4 , 1 , - 7 ] , [ tizenöt , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )nyomtatás ( Tens2 )
Ezzel létrejött egy 2D tenzor, amint az alább látható:
3. lépés: Keresse meg a maximális értékű indexeket
Most keresse meg a maximális érték indexét a ' Tens2 ' tenzor a ' torch.argmax() ” funkció:
T2_ind = fáklya. argmax ( Tens2 )4. lépés: Nyomtassa ki a maximális érték indexét
Végül jelenítse meg a maximális érték indexét a bemeneti tenzorban:
nyomtatás ( 'Indexek:' , T2_ind )Az alábbi kimenet szerint a maximális érték indexe a „ Tens2 ” tenzor a „3”. Ez azt jelenti, hogy a tenzor legmagasabb értéke a 3. indexnél van, amely ' tizenöt ”:
5. lépés: Keresse meg a maximális érték indexeit az oszlopok mentén
Ezenkívül a felhasználók a tenzor minden oszlopában megtalálhatják a maximális értékek indexeit/indexeit. Használhatjuk például a „ dim=0 ” argumentum a „torch.argmax()” függvénnyel. Megkeresi a maximális értékek indexeit a „ Tens2 ” tenzorral, majd kiírja ezeket az indexeket:
col_index = fáklya. argmax ( Tens2 , homályos = 0 )nyomtatás ( 'Indexek oszlopokban:' , col_index )
Az alábbi kimenet a maximális értékek indexeit mutatja a tenzor egyes oszlopai mentén:
6. lépés: Keresse meg a maximális érték indexeit a sorok mentén
Hasonlóképpen, a felhasználók megtalálhatják a maximális értékek indexeit/indexeit a tenzor egyes sorai mentén. Például használja a „ dim=1 ” argumentum a „torch.argmax()” függvénnyel, hogy megkeresse a maximális értékek indexeit a „Tens2” tenzor sorai mentén, majd kinyomtassa ezeket az indexeket:
sor_index = fáklya. argmax ( Tens2 , homályos = 1 )nyomtatás ( 'Indexek a sorokban:' , sor_index )
A „Tens2” tenzor egyes sorai mentén a maximális érték indexei az alábbiakban láthatók:
Hatékonyan elmagyaráztuk a PyTorch „torch.argmax()” metódusának használatának módszerét.
jegyzet : Google Colab jegyzetfüzetünket ezen a címen érheti el link .
Következtetés
A PyTorch „torch.argmax()” metódusának használatához először importálja a „ fáklya ” könyvtár. Ezután hozza létre a kívánt 1D vagy 2D tenzort, és tekintse meg annak elemeit. Ezután használja a „ torch.argmax() ” módszer a tenzor maximális értékeinek indexeinek/indexeinek megkeresésére/számítására. Ezenkívül a felhasználók a tenzor minden sorában vagy oszlopában megtalálhatják a maximális érték indexeit a „ homályos ” érvelést. Végül jelenítse meg a maximális érték indexét a bemeneti tenzorban. Ez a blog példát mutatott be a PyTorch „torch.argmax()” metódusának használatára.