Hvordan bruger man Expand Operation i PyTorch?

PyTorch er et maskinlæringsbibliotek, der tillader brugere at arbejde med tensorer. Tensorer er essentielle datastrukturer, der kan have forskellige typer data og former. Nogle gange vil brugere måske udføre udvidelsesoperationen på tensorer for at udvide deres størrelse. Udvid-operationen gør det muligt for brugere at gentage en tensor langs specificerede dimensioner. PyTorch giver 'expand()' attributten, der tager en tensor og en liste over størrelser som input. Det returnerer en ny tensor, der har de samme data, men med forskellige dimensioner.

Denne artikel vil illustrere metoden til at bruge udvidelsesoperationen på tensorer i PyTorch.

Hvordan bruger man Expand Operation i PyTorch?

For at bruge udvidelsesoperationen i PyTorch, tjek nedenstående trin:

• Importer PyTorch-bibliotek
• Opret en ønsket tensor
• Se inputtensorstørrelse
• Udvid tensoren ved hjælp af ' udvide() ' attribut
• Vis udvidet tensor og dens størrelse

Trin 1: Importer PyTorch Library

Først skal du importere ' fakkel ”-bibliotek for at bruge udvidelsesoperationen:

Trin 2: Opret en Tensor

Opret derefter en ønsket tensor ved hjælp af ' torch.tensor() ”-funktion og udskriv dens elementer. Her laver vi følgende ' Tiere ' tensor:

I nedenstående output er tensoren blevet oprettet med succes:

Trin 3: Se inputtensorstørrelse

Se derefter størrelsen på den ovenfor oprettede ' Tiere ' tensor ved hjælp af ' størrelse() ' attribut:

Ifølge nedenstående output er størrelsen af ​​tensoren 3×1:

Trin 4: Udvid Tensor

Brug nu ' udvide() ” attribut for at udføre ekspansionsoperationen og udvide tensoren til en ny dimension. Det er påkrævet at angive tensoren og listen over størrelser som input. Her udvider vi tensoren til 3×4 størrelse:

Trin 5: Vis udvidet tensor og dens størrelse

Til sidst skal du udskrive den udvidede tensor's elementer og dens størrelse:

Nedenstående output viser den udvidede tensor og dens størrelse, dvs. 3×4. Dette indikerer, at den udvidede handling er blevet udført med succes:

Vi har forklaret den effektive metode til at bruge udvidelsesoperationen i PyTorch

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

Konklusion

For at bruge udvidelsesoperationen i PyTorch skal du først importere fakkelbiblioteket. Opret derefter den ønskede tensor og se dens elementer og størrelse. Brug derefter ' udvide() ” attribut for at udvide inputtensoren. Til sidst skal du udskrive den udvidede tensor og se dens størrelse. Denne artikel har demonstreret metoden til at bruge udvidelsesoperationen på tensorer i PyTorch.