Hvordan tilføjer man en dimension til en tensor i PyTorch?

Denne artikel vil eksemplificere metoden til at tilføje dimensioner til en Tensor i PyTorch.

Hvordan tilføjer man en dimension til den specifikke tensor i PyTorch?

Brugere kan tilføje dimensioner til enhver Tensor, såsom en 1D-tensor eller 2D-tensor i PyTorch. For at tilføje nye dimensioner til tensorer på en bestemt position, tjek følgende eksempler for en bedre forståelse:

• Eksempel 1: Tilføj dimension til en 1D-tensor i PyTorch
• Eksempel 2: Tilføj dimension til en 2D-tensor i PyTorch

Eksempel 1: Tilføj dimension til en 1D-tensor i PyTorch

I dette eksempel vil vi oprette en 1D-tensor og tilføje en dimension til den på en bestemt position. Følg nedenstående trin for en praktisk demonstration:

Trin 1: Importer bibliotek
Importer først fakkelbiblioteket:

Trin 2: Opret 1D Tensor
Opret derefter en endimensionel tensor. For eksempel har vi oprettet følgende tensor og gemt den i ' x variabel:

Trin 3: Se Tensor Shape
Vis derefter den nyoprettede tensorform for at se dens dimensioner:

Nedenstående output indikerer, at vores tensor er endimensionel:

Trin 4: Tilføj Dimension til 1D Tensor
Brug nu ' torch.unsqueeze(input, dæmp) ”-funktion for at tilføje dimensionen til 1D-tensoren ved den specifikke position. For eksempel tilføjer vi dimensionen til tensoren ved 0-indekset:

Her,

• “ nye_tiere ” er den variabel, der inkluderer den tilføjede dimension.
• “ x ” er inputtensoren.
• “ dim=0 ” bruges til at tilføje dimension ved 0-indeks.

Trin 5: Bekræft output
Sørg endelig for, at en ny dimension er blevet tilføjet til tensoren eller ej:

I nedenstående output kan det ses, at den nye dimension er blevet tilføjet til 1D-tensoren ved 0-indekset:

Desuden kan brugere også tilføje dimensioner til andre positioner. Her har vi tilføjet dimensionen ved første indeks:

Eksempel 2: Tilføj dimension til en 2D-tensor i PyTorch

Her vil vi skabe/lave en 2D-tensor og tilføje en dimension til den på den specifikke position. Prøv de givne trin til praktisk implementering:

Trin 1: Importer fakkelbibliotek
Importer først fakkelbiblioteket:

Trin 2: Opret en 2D-tensor
Opret derefter en todimensionel tensor. For eksempel har vi oprettet følgende tensor og gemt den i ' x variabel:

Trin 3: Se Tensor Shape
Derefter skal du vise den nyoprettede tensorform for at se dens dimensioner:

Ifølge nedenstående output er denne tensor todimensionel:

Trin 4: Tilføj Dimension til 2D Tensor
Tilføj nu dimensionen til 2D-tensoren ved den specifikke position ved hjælp af 'torch.unsqueeze(input, dæmp)' fungere. For eksempel tilføjer vi dimensionen til tensoren ved 0-indekset:

Trin 5: Bekræft output
Til sidst skal du kontrollere, om den nye dimension er blevet tilføjet til 2D-tensoren eller ej:

Nedenstående output indikerer, at den nye dimension er blevet tilføjet til 2D-tensoren ved 0-indekset:

Bemærk : Du kan få adgang til vores Google Colab Notebook her link .

Vi har effektivt forklaret metoden til at tilføje dimension til en tensor i PyTorch ved hjælp af forskellige eksempler.

Konklusion

For at tilføje dimensioner til en tensor i PyTorch skal du først importere PyTorch-biblioteket. Opret derefter en 1D- eller 2D-tensor og se dens dimensioner. Tilføj derefter dimensionen til en tensor på den specifikke position ved at bruge ' torch.unsqueeze(input, dæmp) ' funktion. Brugere skal overføre inputtensoren og den ønskede indeksposition som en parameter til denne funktion. Denne artikel har eksemplificeret metoden til at tilføje dimensioner til en tensor i PyTorch.