Ez a cikk példát mutat be, hogyan lehet dimenziókat hozzáadni egy Tensorhoz a PyTorch alkalmazásban.
Hogyan adhatunk dimenziót a PyTorch specifikus tenzorához?
A felhasználók méretet adhatnak bármely tenzorhoz, például 1D tenzorhoz vagy 2D tenzorhoz a PyTorchban. Ha új dimenziókat szeretne hozzáadni a tenzorokhoz egy adott pozícióban, nézze meg a következő példákat a jobb megértés érdekében:
- 1. példa: Dimenzió hozzáadása egy 1D tenzorhoz a PyTorch alkalmazásban
- 2. példa: Dimenzió hozzáadása egy 2D tenzorhoz a PyTorch alkalmazásban
1. példa: Dimenzió hozzáadása egy 1D tenzorhoz a PyTorch alkalmazásban
Ebben a példában létrehozunk egy 1D tenzort, és egy adott helyen dimenziót adunk hozzá. Kövesse az alábbi lépéseket a gyakorlati bemutatóhoz:
1. lépés: Könyvtár importálása
Először importálja a fáklyakönyvtárat:
import fáklya
2. lépés: Hozzon létre 1D tenzort
Ezután hozzon létre egy egydimenziós tenzort. Például létrehoztuk a következő tenzort, és tároltuk a „ x ” változó:
x = fáklya. tenzor ( [ 5 , 3 , 8 , 2 ] )
3. lépés: Tekintse meg a tenzor alakját
Ezután jelenítse meg az újonnan létrehozott tenzor alakzatot a méreteinek megtekintéséhez:
Az alábbi kimenet azt jelzi, hogy a tenzorunk egydimenziós:
4. lépés: Adjon hozzá méretet az 1D tenzorhoz
Most használja a „ torch.unsqueeze(bemenet, dim) ” funkcióval hozzáadhatja a dimenziót az 1D tenzorhoz az adott helyen. Például hozzáadjuk a dimenziót a tenzorhoz 0 indexnél:
Itt,
- ' új_tízek ” az a változó, amely tartalmazza a hozzáadott dimenziót.
- ' x ” a bemeneti tenzor.
- ' dim=0 ” dimenzió hozzáadására szolgál 0 indexnél.
5. lépés: Ellenőrizze a kimenetet
Végül győződjön meg arról, hogy hozzáadott-e új dimenziót a tenzorhoz, vagy sem:
Az alábbi kimenetben megfigyelhető, hogy az új dimenziót hozzáadtuk az 1D tenzorhoz a 0 indexnél:
Sőt, a felhasználók méreteket is hozzáadhatnak más pozíciókhoz. Itt hozzáadtuk a dimenziót az első indexhez:
2. példa: Dimenzió hozzáadása egy 2D tenzorhoz a PyTorch alkalmazásban
Itt létrehozunk/készítünk egy 2D tenzort, és hozzáadunk egy dimenziót az adott helyen. Próbálja ki a megadott lépéseket a gyakorlati megvalósításhoz:
1. lépés: Importálja a Torch Library-t
Először importálja a fáklyakönyvtárat:
2. lépés: Hozzon létre egy 2D tenzort
Ezután hozzon létre egy kétdimenziós tenzort. Például létrehoztuk a következő tenzort, és tároltuk a „ x ” változó:
3. lépés: Tekintse meg a tenzor alakját
Ezután jelenítse meg az újonnan létrehozott tenzor alakzatot a méreteinek megtekintéséhez:
Az alábbi kimenet szerint ez a tenzor kétdimenziós:
4. lépés: Adjon hozzá méretet a 2D tenzorhoz
Most adja hozzá a dimenziót a 2D tenzorhoz az adott helyen a gombbal 'torch.unsqueeze(bemenet, dim)' funkció. Például hozzáadjuk a dimenziót a tenzorhoz 0 indexnél:
5. lépés: Ellenőrizze a kimenetet
Végül ellenőrizze, hogy az új dimenzió hozzáadásra került-e a 2D tenzorhoz vagy sem:
Az alábbi kimenet azt jelzi, hogy az új dimenzió sikeresen hozzáadva a 2D tenzorhoz a 0 indexnél:
jegyzet : Google Colab jegyzetfüzetünket ezen a címen érheti el link .
Különböző példákon keresztül hatékonyan elmagyaráztuk a PyTorch tenzor dimenziójának hozzáadásának módszerét.
Következtetés
Ha dimenziókat szeretne hozzáadni egy tenzorhoz a PyTorch alkalmazásban, először importálja a PyTorch könyvtárat. Ezután hozzon létre egy 1D vagy 2D tenzort, és tekintse meg a méreteit. Ezután adja hozzá a dimenziót egy tenzorhoz az adott pozícióban a ' torch.unsqueeze(bemenet, dim) ” funkciót. A felhasználóknak át kell adniuk a bemeneti tenzort és a kívánt indexpozíciót paraméterként ehhez a funkcióhoz. Ez a cikk példát mutat be a PyTorch tenzorokhoz való dimenziók hozzáadásának módszerével.