PyTorch je knjižnica za strojno učenje, ki uporabnikom omogoča delo s tenzorji. Tenzorji so bistvene podatkovne strukture, ki imajo lahko različne vrste podatkov in oblike. Včasih bodo uporabniki morda želeli izvesti operacijo razširitve na tenzorjih, da povečajo njihovo velikost. Operacija razširitve omogoča uporabnikom ponavljanje tenzorja vzdolž določenih dimenzij. PyTorch ponuja atribut »expand()«, ki kot vhodne podatke sprejme tenzor in seznam velikosti. Vrne nov tenzor z enakimi podatki, vendar z različnimi dimenzijami.
Ta članek bo prikazal metodo za uporabo operacije razširitve na tenzorjih v PyTorchu.
Kako uporabljati operacijo razširitve v PyTorchu?
Če želite uporabiti operacijo razširitve v PyTorchu, si oglejte spodnje korake:
- Uvoz knjižnice PyTorch
- Ustvari želeni tenzor
- Oglejte si velikost vhodnega tenzorja
- Razširite tenzor z uporabo ' razširi() ” atribut
- Prikaz razširjenega tenzorja in njegove velikosti
1. korak: uvozite knjižnico PyTorch
Najprej uvozite » svetilka ” za uporabo operacije razširitve:
uvoz svetilka
2. korak: Ustvarite tenzor
Nato ustvarite želeni tenzor z uporabo ' torch.tensor() ” in natisnite njene elemente. Tukaj ustvarjamo naslednje ' desetice ” tenzor:
desetice = svetilka. tenzor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
tiskanje ( desetice )
V spodnjem izhodu je bil tenzor uspešno ustvarjen:
3. korak: Oglejte si velikost vhodnega tenzorja
Nato si oglejte velikost zgoraj ustvarjenega ' desetice ' tenzor z uporabo ' velikost () ” atribut:
tiskanje ( 'Velikost tenzorja:' , desetice. velikost ( ) )Glede na spodnji rezultat je velikost tenzorja 3×1:
4. korak: Razširite Tensor
Zdaj uporabite » razširi () ” za izvedbo operacije razširitve in razširitev tenzorja na novo dimenzijo. Kot vhod je treba zagotoviti tenzor in seznam velikosti. Tukaj razširimo tenzor na velikost 3×4:
Exp_tens = desetice. razširiti ( 3 , 4 )5. korak: Prikažite razširjeni tenzor in njegovo velikost
Na koncu natisnite elemente razširjenega tenzorja in njegovo velikost:
tiskanje ( Exp_tens )tiskanje ( Exp_tens. velikost ( ) )
Spodnji rezultat prikazuje razširjeni tenzor in njegovo velikost, tj. 3×4. To pomeni, da je bila razširjena operacija uspešno izvedena:
Razložili smo učinkovito metodo za uporabo operacije razširitve v PyTorchu
Opomba : Do našega zvezka Google Colab lahko dostopate tukaj povezava .
Zaključek
Če želite uporabiti operacijo razširitve v PyTorchu, najprej uvozite knjižnico torča. Nato ustvarite želeni tenzor in si oglejte njegove elemente in velikost. Nato uporabite » razširi() ” za razširitev vhodnega tenzorja. Na koncu natisnite razširjeni tenzor in si oglejte njegovo velikost. Ta članek je prikazal metodo za uporabo operacije razširitve na tenzorjih v PyTorchu.