In diesem Artikel wird die Methode zum Hinzufügen von Dimensionen zu einem Tensor in PyTorch veranschaulicht.
Wie füge ich dem spezifischen Tensor in PyTorch eine Dimension hinzu?
Benutzer können jedem Tensor Dimensionen hinzufügen, beispielsweise einem 1D-Tensor oder einem 2D-Tensor in PyTorch. Um Tensoren an einer bestimmten Position neue Dimensionen hinzuzufügen, sehen Sie sich zum besseren Verständnis die folgenden Beispiele an:
- Beispiel 1: Dimension zu einem 1D-Tensor in PyTorch hinzufügen
- Beispiel 2: Dimension zu einem 2D-Tensor in PyTorch hinzufügen
Beispiel 1: Dimension zu einem 1D-Tensor in PyTorch hinzufügen
In diesem Beispiel erstellen wir einen 1D-Tensor und fügen ihm an einer bestimmten Position eine Dimension hinzu. Befolgen Sie die unten aufgeführten Schritte für eine praktische Demonstration:
Schritt 1: Bibliothek importieren
Importieren Sie zunächst die Torch-Bibliothek:
importieren Fackel
Schritt 2: Erstellen Sie einen 1D-Tensor
Erstellen Sie dann einen eindimensionalen Tensor. Zum Beispiel haben wir den folgenden Tensor erstellt und ihn im „ X ” Variable:
X = Fackel. Tensor ( [ 5 , 3 , 8 , 2 ] )
Schritt 3: Tensorform anzeigen
Zeigen Sie als Nächstes die neu erstellte Tensorform an, um ihre Abmessungen anzuzeigen:
Die folgende Ausgabe zeigt, dass unser Tensor eindimensional ist:
Schritt 4: Dimension zum 1D-Tensor hinzufügen
Nutzen Sie nun die „ Torch.unsqueeze(Eingabe, Dimmen) ”Funktion zum Hinzufügen der Dimension zum 1D-Tensor an der bestimmten Position. Zum Beispiel fügen wir die Dimension zum Tensor bei Index 0 hinzu:
Hier,
- „ new_tens „ist die Variable, die die hinzugefügte Dimension enthält.
- „ X „ist der Eingabetensor.
- „ dim=0 „wird verwendet, um eine Dimension bei Index 0 hinzuzufügen.
Schritt 5: Ausgabe überprüfen
Stellen Sie abschließend sicher, dass dem Tensor eine neue Dimension hinzugefügt wurde oder nicht:
In der folgenden Ausgabe ist zu erkennen, dass die neue Dimension dem 1D-Tensor am Index 0 hinzugefügt wurde:
Darüber hinaus können Benutzer auch Bemaßungen zu anderen Positionen hinzufügen. Hier haben wir die Dimension am ersten Index hinzugefügt:
Beispiel 2: Dimension zu einem 2D-Tensor in PyTorch hinzufügen
Hier erstellen/erstellen wir einen 2D-Tensor und fügen ihm an der spezifischen Position eine Dimension hinzu. Probieren Sie die vorgegebenen Schritte zur praktischen Umsetzung aus:
Schritt 1: Torch-Bibliothek importieren
Importieren Sie zunächst die Torch-Bibliothek:
Schritt 2: Erstellen Sie einen 2D-Tensor
Erstellen Sie dann einen zweidimensionalen Tensor. Zum Beispiel haben wir den folgenden Tensor erstellt und ihn im „ X ” Variable:
Schritt 3: Tensorform anzeigen
Zeigen Sie anschließend die neu erstellte Tensorform an, um ihre Abmessungen anzuzeigen:
Gemäß der folgenden Ausgabe ist dieser Tensor zweidimensional:
Schritt 4: Dimension zum 2D-Tensor hinzufügen
Fügen Sie nun die Dimension mithilfe von zum 2D-Tensor an der bestimmten Position hinzu „torch.unsqueeze(input, dim)“ Funktion. Zum Beispiel fügen wir die Dimension zum Tensor bei Index 0 hinzu:
Schritt 5: Ausgabe überprüfen
Überprüfen Sie abschließend, ob die neue Dimension zum 2D-Tensor hinzugefügt wurde oder nicht:
Die folgende Ausgabe zeigt an, dass die neue Dimension erfolgreich zum 2D-Tensor am Index 0 hinzugefügt wurde:
Notiz : Hier können Sie auf unser Google Colab Notebook zugreifen Verknüpfung .
Wir haben die Methode zum Hinzufügen einer Dimension zu einem Tensor in PyTorch anhand verschiedener Beispiele effizient erklärt.
Abschluss
Um einem Tensor in PyTorch Dimensionen hinzuzufügen, importieren Sie zunächst die PyTorch-Bibliothek. Erstellen Sie dann einen 1D- oder 2D-Tensor und sehen Sie sich seine Abmessungen an. Fügen Sie anschließend die Dimension zu einem Tensor an der bestimmten Position hinzu, indem Sie „ Torch.unsqueeze(Eingabe, Dimmen) ” Funktion. Benutzer müssen den Eingabetensor und die gewünschte Indexposition als Parameter an diese Funktion übergeben. In diesem Artikel wurde die Methode zum Hinzufügen von Dimensionen zu einem Tensor in PyTorch veranschaulicht.