Hvordan bruke Expand Operation i PyTorch?

PyTorch er et maskinlæringsbibliotek som lar brukere jobbe med tensorer. Tensorer er essensielle datastrukturer som kan ha forskjellige typer data og former. Noen ganger vil brukere kanskje utføre utvidelsesoperasjonen på tensorer for å utvide størrelsen. Utvidelsesoperasjonen gjør det mulig for brukere å gjenta en tensor langs spesifiserte dimensjoner. PyTorch gir «expand()»-attributtet som tar en tensor og en liste over størrelser som innganger. Den returnerer en ny tensor som har samme data, men med forskjellige dimensjoner.

Denne artikkelen vil illustrere metoden for å bruke utvidelsesoperasjonen på tensorer i PyTorch.

Hvordan bruke Expand Operation i PyTorch?

For å bruke utvidelsesoperasjonen i PyTorch, sjekk ut trinnene nedenfor:

• Importer PyTorch-biblioteket
• Lag en ønsket tensor
• Se inndatatensorstørrelse
• Utvid tensoren ved å bruke ' utvide() ' Egenskap
• Vis utvidet tensor og dens størrelse

Trinn 1: Importer PyTorch-biblioteket

Først importerer du ' lommelykt ”-biblioteket for å bruke utvidelsesoperasjonen:

Trinn 2: Lag en tensor

Deretter oppretter du en ønsket tensor ved å bruke ' torch.tensor() ”-funksjonen og skrive ut elementene. Her lager vi følgende ' Tiere ' tensor:

I utgangen nedenfor er tensoren opprettet med hell:

Trinn 3: Se Input Tensor Size

Deretter kan du se størrelsen på den ovenfor opprettede ' Tiere ' tensor ved å bruke ' størrelse() ' Egenskap:

I henhold til utgangen nedenfor er størrelsen på tensoren 3×1:

Trinn 4: Utvid Tensor

Bruk nå ' utvide() ”-attributt for å utføre utvidelsesoperasjonen og utvide tensoren til en ny dimensjon. Det er nødvendig å oppgi tensoren og listen over størrelser som input. Her utvider vi tensoren til 3×4 størrelse:

Trinn 5: Vis utvidet tensor og dens størrelse

Til slutt, skriv ut den utvidede tensorens elementer og størrelsen:

Utgangen nedenfor viser den utvidede tensoren og dens størrelse, dvs. 3×4. Dette indikerer at den utvidede operasjonen har blitt utført vellykket:

Vi har forklart den effektive metoden for å bruke utvidelsesoperasjonen i PyTorch

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

Konklusjon

For å bruke utvidelsesoperasjonen i PyTorch, importer først fakkelbiblioteket. Lag deretter ønsket tensor og se elementene og størrelsen. Deretter bruker du ' utvide() ”-attributt for å utvide inngangstensoren. Til slutt, skriv ut den utvidede tensoren og se størrelsen. Denne artikkelen har demonstrert metoden for å bruke utvidelsesoperasjonen på tensorer i PyTorch.