PyTorch este o bibliotecă de învățare automată care permite utilizatorilor să lucreze cu tensori. Tensorii sunt structuri de date esențiale care pot avea diferite tipuri de date și forme. Uneori, utilizatorii pot dori să efectueze operația de extindere pe tensori pentru a le extinde dimensiunea. Operația de extindere permite utilizatorilor să repete un tensor de-a lungul dimensiunilor specificate. PyTorch oferă atributul „expand()” care ia ca intrări un tensor și o listă de dimensiuni. Returnează un nou tensor care are aceleași date, dar cu dimensiuni diferite.
Acest articol va ilustra metoda de utilizare a operației de extindere pe tensori în PyTorch.
Cum se utilizează operația de extindere în PyTorch?
Pentru a utiliza operația de extindere în PyTorch, consultați pașii furnizați mai jos:
- Importați biblioteca PyTorch
- Creați un tensor dorit
- Vedeți dimensiunea tensorului de intrare
- Extinde tensorul folosind „ extinde() ” atribut
- Afișează tensorul extins și dimensiunea acestuia
Pasul 1: importați biblioteca PyTorch
Mai întâi, importați „ torță ” bibliotecă pentru a utiliza operația de extindere:
import torță
Pasul 2: Creați un tensor
Apoi, creați un tensor dorit folosind „ torță.tensor() ” funcţionează şi imprimă elementele acestuia. Aici, creăm următorul „ Zeci ” tensor:
Zeci = torță. tensor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
imprimare ( Zeci )
În rezultatul de mai jos, tensorul a fost creat cu succes:
Pasul 3: Vedeți dimensiunea tensorului de intrare
Apoi, vizualizați dimensiunea „ Zeci ” tensorul folosind “ mărimea() ” atribut:
imprimare ( „Dimensiunea tensorului:” , Zeci. mărimea ( ) )Conform rezultatului de mai jos, dimensiunea tensorului este 3×1:
Pasul 4: Extindeți Tensor
Acum, folosiți „ extinde() ” pentru a efectua operația de extindere și extinde tensorul la o nouă dimensiune. Este necesar să se furnizeze tensorul și lista de dimensiuni ca intrare. Aici, extindem tensorul la dimensiunea 3×4:
Exp_zeci = Zeci. extinde ( 3 , 4 )Pasul 5: Afișați Tensorul extins și dimensiunea acestuia
În cele din urmă, imprimați elementele tensorului extins și dimensiunea acestuia:
imprimare ( Exp_zeci )imprimare ( Exp_zeci. mărimea ( ) )
Ieșirea de mai jos arată tensorul extins și dimensiunea acestuia, adică 3×4. Aceasta indică faptul că operația extinsă a fost efectuată cu succes:
Am explicat metoda eficientă de utilizare a operațiunii de extindere în PyTorch
Notă : Puteți accesa blocnotesul nostru Google Colab aici legătură .
Concluzie
Pentru a utiliza operația de extindere în PyTorch, importați mai întâi biblioteca torță. Apoi, creați tensorul dorit și vizualizați elementele și dimensiunea acestuia. Apoi, utilizați „ extinde() ” atribut pentru a extinde tensorul de intrare. În cele din urmă, imprimați tensorul extins și vizualizați dimensiunea acestuia. Acest articol a demonstrat metoda de utilizare a operației de extindere pe tensori în PyTorch.