PyTorch je knižnica strojového učenia, ktorá používateľom umožňuje pracovať s tenzormi. Tenzory sú základné dátové štruktúry, ktoré môžu mať rôzne typy dát a tvarov. Niekedy môžu používatelia chcieť vykonať operáciu rozšírenia na tenzoroch, aby zväčšili ich veľkosť. Operácia rozbalenia umožňuje používateľom opakovať tenzor pozdĺž špecifikovaných rozmerov. PyTorch poskytuje atribút „expand()“, ktorý berie ako vstupy tenzor a zoznam veľkostí. Vracia nový tenzor, ktorý má rovnaké dáta, ale s inými rozmermi.
Tento článok ilustruje metódu použitia operácie expandovania na tenzoroch v PyTorch.
Ako používať Expand Operation v PyTorch?
Ak chcete použiť operáciu rozšírenia v PyTorch, pozrite si nižšie uvedené kroky:
- Importujte knižnicu PyTorch
- Vytvorte požadovaný tenzor
- Zobraziť veľkosť vstupného tenzora
- Rozbaľte tenzor pomocou „ expand() “
- Zobrazte rozšírený tenzor a jeho veľkosť
Krok 1: Importujte knižnicu PyTorch
Najprv importujte súbor „ fakľa ” knižnica na použitie operácie rozbalenia:
importovať fakľa
Krok 2: Vytvorte tenzor
Potom vytvorte požadovaný tenzor pomocou „ pochodeň.tensor() ” a vytlačte jej prvky. Tu vytvárame nasledujúce „ Desiatky “tensor:
Desiatky = fakľa. tenzor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
vytlačiť ( Desiatky )
V nižšie uvedenom výstupe bol tenzor úspešne vytvorený:
Krok 3: Zobrazte veľkosť vstupného tenzora
Ďalej si pozrite veľkosť vyššie vytvoreného „ Desiatky “tensor pomocou “ veľkosť () ” atribút:
vytlačiť ( 'Veľkosť tenzora:' , Desiatky. veľkosť ( ) )Podľa nižšie uvedeného výstupu je veľkosť tenzora 3 × 1:
Krok 4: Rozbaľte Tensor
Teraz použite „ expand() ” na vykonanie operácie expandovania a rozšírenie tenzora do novej dimenzie. Ako vstup je potrebné poskytnúť tenzor a zoznam veľkostí. Tu rozširujeme tenzor na veľkosť 3×4:
Exp_des = Desiatky. expandovať ( 3 , 4 )Krok 5: Zobrazte rozšírený tenzor a jeho veľkosť
Nakoniec vytlačte prvky expandovaného tenzora a jeho veľkosť:
vytlačiť ( Exp_des )vytlačiť ( Exp_des. veľkosť ( ) )
Nižšie uvedený výstup zobrazuje rozšírený tenzor a jeho veľkosť, tj 3×4. To znamená, že operácia rozšírenia bola úspešne vykonaná:
Vysvetlili sme efektívnu metódu použitia operácie rozšírenia v PyTorch
Poznámka : K nášmu Zápisníku Google Colab môžete pristupovať tu odkaz .
Záver
Ak chcete použiť operáciu rozbalenia v PyTorch, najprv importujte knižnicu horáka. Potom vytvorte požadovaný tenzor a zobrazte jeho prvky a veľkosť. Ďalej použite „ expand() ” na rozšírenie vstupného tenzora. Nakoniec vytlačte rozšírený tenzor a zobrazte jeho veľkosť. Tento článok demonštroval metódu použitia operácie expandovania na tenzoroch v PyTorch.