PyTorch je biblioteka za strojno učenje koja korisnicima omogućuje rad s tenzorima. Tenzori su bitne strukture podataka koje mogu imati različite vrste podataka i oblika. Ponekad će korisnici možda htjeti izvršiti operaciju proširivanja na tenzorima kako bi proširili njihovu veličinu. Operacija proširivanja omogućuje korisnicima ponavljanje tenzora duž navedenih dimenzija. PyTorch pruža atribut 'expand()' koji kao ulaz uzima tenzor i popis veličina. Vraća novi tenzor koji ima iste podatke, ali s različitim dimenzijama.
Ovaj će članak ilustrirati metodu korištenja operacije proširenja na tenzorima u PyTorchu.
Kako koristiti operaciju proširenja u PyTorchu?
Da biste koristili operaciju proširivanja u PyTorchu, pogledajte dolje navedene korake:
- Uvoz PyTorch biblioteke
- Napravite željeni tenzor
- Pogledajte veličinu tenzora unosa
- Proširite tenzor koristeći ' proširiti() ” atribut
- Prikaz proširenog tenzora i njegove veličine
Korak 1: Uvezite PyTorch biblioteku
Prvo uvezite ' baklja ” za korištenje operacije proširivanja:
uvoz baklja
Korak 2: Napravite tenzor
Zatim kreirajte željeni tenzor koristeći ' baklja.tenzor() ” funkciju i ispisati njezine elemente. Ovdje stvaramo sljedeće ' desetice ” tenzor:
desetice = baklja. tenzor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
ispisati ( desetice )
U donjem izlazu, tenzor je uspješno kreiran:
Korak 3: Pregledajte veličinu ulaznog tenzora
Zatim pogledajte veličinu gore stvorenog ' desetice ' tenzor koristeći ' veličina() ” atribut:
ispisati ( 'Veličina tenzora:' , desetice. veličina ( ) )Prema izlazu ispod, veličina tenzora je 3×1:
Korak 4: Proširite Tensor
Sada upotrijebite ' proširiti() ” za izvođenje operacije proširenja i proširenje tenzora na novu dimenziju. Potrebno je dati tenzor i popis veličina kao ulaz. Ovdje proširujemo tenzor na veličinu 3×4:
Exp_tens = desetice. proširiti ( 3 , 4 )Korak 5: Prikažite prošireni tenzor i njegovu veličinu
Na kraju ispišite elemente proširenog tenzora i njegovu veličinu:
ispisati ( Exp_tens )ispisati ( Exp_tens. veličina ( ) )
Donji rezultat prikazuje prošireni tenzor i njegovu veličinu, tj. 3×4. Ovo označava da je proširena operacija uspješno izvedena:
Objasnili smo učinkovitu metodu korištenja operacije proširenja u PyTorchu
Bilješka : Našoj Google Colab bilježnici možete pristupiti ovdje veza .
Zaključak
Da biste koristili operaciju proširivanja u PyTorchu, prvo uvezite torch biblioteku. Zatim izradite željeni tenzor i pogledajte njegove elemente i veličinu. Zatim upotrijebite ' proširiti() ” za proširenje ulaznog tenzora. Na kraju ispišite prošireni tenzor i pogledajte njegovu veličinu. Ovaj je članak pokazao metodu korištenja operacije proširenja na tenzorima u PyTorchu.