Hvordan legge til en dimensjon til en tensor i PyTorch?

Denne artikkelen vil eksemplifisere metoden for å legge til dimensjoner til en Tensor i PyTorch.

Hvordan legge til en dimensjon til den spesifikke tensoren i PyTorch?

Brukere kan legge til dimensjoner til enhver Tensor, for eksempel en 1D-tensor eller 2D-tensor i PyTorch. For å legge til nye dimensjoner til tensorer på en bestemt posisjon, sjekk ut følgende eksempler for en bedre forståelse:

• Eksempel 1: Legg til dimensjon til en 1D-tensor i PyTorch
• Eksempel 2: Legg til dimensjon til en 2D-tensor i PyTorch

Eksempel 1: Legg til dimensjon til en 1D-tensor i PyTorch

I dette eksemplet vil vi lage en 1D-tensor og legge til en dimensjon til den på en bestemt posisjon. Følg trinnene nedenfor for en praktisk demonstrasjon:

Trinn 1: Importer bibliotek
Først importerer du fakkelbiblioteket:

Trinn 2: Lag 1D Tensor
Deretter lager du en endimensjonal tensor. For eksempel har vi laget følgende tensor og lagret den i ' x variabel:

Trinn 3: Se Tensor Shape
Deretter viser du den nyopprettede tensorformen for å se dens dimensjoner:

Utgangen nedenfor indikerer at tensoren vår er endimensjonal:

Trinn 4: Legg til dimensjon til 1D Tensor
Bruk nå ' torch.unsqueeze(input, dim) ”-funksjon for å legge til dimensjonen til 1D-tensoren ved den spesifikke posisjonen. For eksempel legger vi til dimensjonen til tensoren ved 0-indeksen:

Her,

• ' nye_tiere ” er variabelen som inkluderer den ekstra dimensjonen.
• ' x ” er inngangstensoren.
• ' dim=0 ' brukes til å legge til dimensjon ved 0-indeks.

Trinn 5: Bekreft utdata
Til slutt, sørg for at en ny dimensjon er lagt til tensoren eller ikke:

I utgangen nedenfor kan det observeres at den nye dimensjonen er lagt til 1D-tensoren ved 0-indeksen:

Dessuten kan brukere også legge til dimensjoner til andre posisjoner. Her har vi lagt til dimensjonen ved første indeks:

Eksempel 2: Legg til dimensjon til en 2D-tensor i PyTorch

Her vil vi lage/lage en 2D-tensor og legge til en dimensjon til den på den spesifikke posisjonen. Prøv de gitte trinnene for praktisk implementering:

Trinn 1: Importer Torch Library
Først importerer du fakkelbiblioteket:

Trinn 2: Lag en 2D-tensor
Deretter lager du en todimensjonal tensor. For eksempel har vi laget følgende tensor og lagret den i ' x variabel:

Trinn 3: Se Tensor Shape
Deretter viser du den nyopprettede tensorformen for å se dens dimensjoner:

I følge utgangen nedenfor er denne tensoren todimensjonal:

Trinn 4: Legg til dimensjon til 2D Tensor
Legg nå til dimensjonen til 2D-tensoren ved den spesifikke posisjonen ved å bruke «torch.unsqueeze(input, dim)» funksjon. For eksempel legger vi til dimensjonen til tensoren ved 0-indeksen:

Trinn 5: Bekreft utdata
Til slutt, kontroller om den nye dimensjonen er lagt til 2D-tensoren eller ikke:

Utdataene nedenfor indikerer at den nye dimensjonen har blitt lagt til 2D-tensoren ved 0-indeksen:

Merk : Du kan få tilgang til Google Colab Notebook på denne link .

Vi har effektivt forklart metoden for å legge til dimensjoner til en tensor i PyTorch ved å bruke forskjellige eksempler.

Konklusjon

For å legge til dimensjoner til en tensor i PyTorch, importer først PyTorch-biblioteket. Lag deretter en 1D- eller 2D-tensor og se dimensjonene. Deretter legger du dimensjonen til en tensor på den spesifikke posisjonen ved å bruke ' torch.unsqueeze(input, dim) ' funksjon. Brukere må sende inn-tensoren og ønsket indeksposisjon som en parameter til denne funksjonen. Denne artikkelen har eksemplifisert metoden for å legge til dimensjoner til en tensor i PyTorch.