Dit artikel illustreert de methode voor het toevoegen van dimensies aan een Tensor in PyTorch.
Hoe voeg ik een dimensie toe aan de specifieke tensor in PyTorch?
Gebruikers kunnen dimensies toevoegen aan elke Tensor, zoals een 1D-tensor of 2D-tensor in PyTorch. Als u nieuwe dimensies aan tensoren op een specifieke positie wilt toevoegen, bekijkt u de volgende voorbeelden voor een beter begrip:
- Voorbeeld 1: dimensie toevoegen aan een 1D-tensor in PyTorch
- Voorbeeld 2: dimensie toevoegen aan een 2D-tensor in PyTorch
Voorbeeld 1: dimensie toevoegen aan een 1D-tensor in PyTorch
In dit voorbeeld maken we een 1D-tensor en voegen daar op een bepaalde positie een dimensie aan toe. Volg de onderstaande stappen voor een praktische demonstratie:
Stap 1: Bibliotheek importeren
Importeer eerst de fakkelbibliotheek:
importeren fakkel
Stap 2: Maak een 1D-tensor
Maak vervolgens een eendimensionale tensor. We hebben bijvoorbeeld de volgende tensor gemaakt en opgeslagen in de “ X ”variabele:
X = fakkel. tensor ( [ 5 , 3 , 8 , 2 ] )
Stap 3: Bekijk de tensorvorm
Geef vervolgens de nieuw gemaakte tensorvorm weer om de afmetingen ervan te bekijken:
De onderstaande uitvoer geeft aan dat onze tensor eendimensionaal is:
Stap 4: Voeg dimensie toe aan 1D-tensor
Maak nu gebruik van de “ fakkel.unsqueeze(invoer, dimmen) ”-functie om de dimensie toe te voegen aan de 1D-tensor op de specifieke positie. We voegen bijvoorbeeld de dimensie toe aan de tensor op index 0:
Hier,
- “ nieuwe_tens ” is de variabele die de toegevoegde dimensie bevat.
- “ X ” is de ingangstensor.
- “ dimmen=0 ” wordt gebruikt om dimensie toe te voegen aan index 0.
Stap 5: Controleer de uitvoer
Zorg er ten slotte voor dat er al dan niet een nieuwe dimensie aan de tensor is toegevoegd:
In de onderstaande uitvoer kan worden waargenomen dat de nieuwe dimensie is toegevoegd aan de 1D-tensor bij de 0-index:
Bovendien kunnen gebruikers ook dimensies toevoegen aan andere posities. Hier hebben we de dimensie bij de eerste index toegevoegd:
Voorbeeld 2: dimensie toevoegen aan een 2D-tensor in PyTorch
Hier gaan we een 2D-tensor maken/maken en er op de specifieke positie een dimensie aan toevoegen. Probeer de gegeven stappen voor praktische implementatie:
Stap 1: Importeer de fakkelbibliotheek
Importeer eerst de fakkelbibliotheek:
Stap 2: Maak een 2D-tensor
Maak vervolgens een tweedimensionale tensor. We hebben bijvoorbeeld de volgende tensor gemaakt en opgeslagen in de “ X ”variabele:
Stap 3: Bekijk de tensorvorm
Geef daarna de nieuw gemaakte tensorvorm weer om de afmetingen ervan te bekijken:
Volgens de onderstaande uitvoer is deze tensor tweedimensionaal:
Stap 4: Voeg dimensie toe aan 2D-tensor
Voeg nu de dimensie toe aan de 2D-tensor op de specifieke positie met behulp van de “fakkel.unsqueeze(invoer, dimmen)” functie. We voegen bijvoorbeeld de dimensie toe aan de tensor op index 0:
Stap 5: Controleer de uitvoer
Controleer ten slotte of de nieuwe dimensie al dan niet aan de 2D-tensor is toegevoegd:
De onderstaande uitvoer geeft aan dat de nieuwe dimensie met succes is toegevoegd aan de 2D-tensor bij de 0-index:
Opmerking : Hier heeft u toegang tot ons Google Colab Notebook koppeling .
We hebben de methode voor het toevoegen van dimensie aan een tensor in PyTorch efficiënt uitgelegd met behulp van verschillende voorbeelden.
Conclusie
Om dimensies aan een tensor in PyTorch toe te voegen, importeert u eerst de PyTorch-bibliotheek. Maak vervolgens een 1D- of 2D-tensor en bekijk de afmetingen ervan. Voeg daarna de dimensie toe aan een tensor op de specifieke positie met behulp van de “ fakkel.unsqueeze(invoer, dimmen) ” functie. Gebruikers moeten de invoertensor en de gewenste indexpositie als parameter aan deze functie doorgeven. Dit artikel illustreert de methode voor het toevoegen van dimensies aan een tensor in PyTorch.