PyTorch je knihovna pro strojové učení, která uživatelům umožňuje pracovat s tenzory. Tenzory jsou základní datové struktury, které mohou mít různé typy dat a tvarů. Někdy mohou uživatelé chtít provést operaci rozšíření na tenzorech, aby se zvětšila jejich velikost. Operace roztažení umožňuje uživatelům opakovat tenzor podél zadaných rozměrů. PyTorch poskytuje atribut „expand()“, který má jako vstupy tenzor a seznam velikostí. Vrací nový tenzor, který má stejná data, ale s jinými rozměry.
Tento článek bude ilustrovat metodu použití operace expand na tenzorech v PyTorch.
Jak používat Expand Operation v PyTorch?
Chcete-li použít operaci rozbalení v PyTorch, podívejte se na níže uvedené kroky:
- Import knihovny PyTorch
- Vytvořte požadovaný tenzor
- Zobrazit velikost vstupního tenzoru
- Rozbalte tenzor pomocí „ rozšířit() ' atribut
- Zobrazit rozšířený tenzor a jeho velikost
Krok 1: Import knihovny PyTorch
Nejprve importujte „ pochodeň ” knihovny pro použití operace rozbalení:
import pochodeň
Krok 2: Vytvořte tenzor
Poté vytvořte požadovaný tenzor pomocí „ pochodeň.tensor() ” a vytisknout její prvky. Zde vytváříme následující „ Desítky “tensor:
Desítky = pochodeň. tenzor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
tisk ( Desítky )
V níže uvedeném výstupu byl tenzor úspěšně vytvořen:
Krok 3: Zobrazte velikost vstupního tenzoru
Dále si prohlédněte velikost výše vytvořeného „ Desítky 'tensor pomocí ' velikost() ' atribut:
tisk ( 'Velikost tenzoru:' , Desítky. velikost ( ) )Podle níže uvedeného výstupu je velikost tenzoru 3×1:
Krok 4: Rozbalte Tensor
Nyní použijte „ rozšířit() ” pro provedení operace rozbalení a rozšíření tenzoru do nové dimenze. Jako vstup je nutné zadat tenzor a seznam velikostí. Zde rozšiřujeme tenzor na velikost 3×4:
Exp_desítky = Desítky. rozšířit ( 3 , 4 )Krok 5: Zobrazte rozbalený tenzor a jeho velikost
Nakonec vytiskněte prvky expandovaného tenzoru a jeho velikost:
tisk ( Exp_desítky )tisk ( Exp_desítky. velikost ( ) )
Níže uvedený výstup ukazuje rozšířený tenzor a jeho velikost, tj. 3×4. To znamená, že operace rozšíření byla úspěšně provedena:
Vysvětlili jsme efektivní metodu použití operace rozšíření v PyTorch
Poznámka : Zde můžete přistupovat k našemu Zápisníku Google Colab odkaz .
Závěr
Chcete-li použít operaci rozbalení v PyTorch, nejprve importujte knihovnu pochodně. Poté vytvořte požadovaný tenzor a zobrazte jeho prvky a velikost. Dále použijte „ rozšířit() ” pro rozšíření vstupního tenzoru. Nakonec rozbalený tenzor vytiskněte a zobrazte jeho velikost. Tento článek demonstroval metodu použití operace expand na tenzorech v PyTorch.