PyTorch är ett maskinlärande bibliotek som tillåter användare att arbeta med tensorer. Tensorer är viktiga datastrukturer som kan ha olika typer av data och former. Ibland kanske användare vill utföra expansionsoperationen på tensorer för att utöka deras storlek. Expansionsoperationen gör det möjligt för användare att upprepa en tensor längs specificerade dimensioner. PyTorch tillhandahåller attributet 'expand()' som tar en tensor och en lista över storlekar som indata. Den returnerar en ny tensor som har samma data men med olika dimensioner.
Den här artikeln kommer att illustrera metoden för att använda expanderingsoperationen på tensorer i PyTorch.
Hur använder man Expand Operation i PyTorch?
För att använda expanderingsoperationen i PyTorch, kolla in stegen nedan:
- Importera PyTorch-bibliotek
- Skapa en önskad tensor
- Visa ingångstensorstorlek
- Expandera tensorn med hjälp av ' bygga ut() ' attribut
- Visa utökad tensor och dess storlek
Steg 1: Importera PyTorch Library
Importera först ' fackla ”-bibliotek för att använda expanderingsoperationen:
importera fackla
Steg 2: Skapa en tensor
Skapa sedan en önskad tensor med hjälp av ' torch.tensor() ”-funktion och skriv ut dess element. Här skapar vi följande ' Tiotal ' tensor:
Tiotal = fackla. tensor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
skriva ut ( Tiotal )
I utgången nedan har tensorn skapats framgångsrikt:
Steg 3: Visa indatatensorstorlek
Se sedan storleken på ovan skapade ' Tiotal ' tensor med hjälp av ' storlek() ' attribut:
skriva ut ( 'Storlek på Tensor:' , Tiotal. storlek ( ) )Enligt utgången nedan är storleken på tensorn 3×1:
Steg 4: Expandera Tensor
Använd nu ' bygga ut() ”-attribut för att utföra expansionsoperationen och utöka tensorn till en ny dimension. Det är nödvändigt att tillhandahålla tensorn och listan över storlekar som indata. Här utökar vi tensorn till 3×4 storlek:
Exp_tens = Tiotal. bygga ut ( 3 , 4 )Steg 5: Visa utökad tensor och dess storlek
Skriv slutligen ut den utökade tensorens element och dess storlek:
skriva ut ( Exp_tens )skriva ut ( Exp_tens. storlek ( ) )
Utdata nedan visar den utökade tensorn och dess storlek, dvs 3×4. Detta indikerar att den utökade operationen har utförts framgångsrikt:
Vi har förklarat den effektiva metoden för att använda expanderingsoperationen i PyTorch
Notera : Du kan komma åt vår Google Colab Notebook här länk .
Slutsats
För att använda expanderingsoperationen i PyTorch, importera först ficklampsbiblioteket. Skapa sedan önskad tensor och se dess element och storlek. Använd sedan ' bygga ut() ”-attribut för att utöka ingångstensorn. Slutligen, skriv ut den utökade tensorn och se dess storlek. Den här artikeln har demonstrerat metoden för att använda expanderingsoperationen på tensorer i PyTorch.