PyTorch on koneoppiva kirjasto, jonka avulla käyttäjät voivat työskennellä tensoreiden kanssa. Tensorit ovat olennaisia tietorakenteita, joilla voi olla erityyppisiä tietoja ja muotoja. Joskus käyttäjät saattavat haluta suorittaa laajennustoiminnon tensoreille laajentaakseen niiden kokoa. Laajennustoiminnon avulla käyttäjät voivat toistaa tensorin määritetyillä mitoilla. PyTorch tarjoaa 'expand()' -attribuutin, joka ottaa syötteinä tensorin ja kokoluettelon. Se palauttaa uuden tensorin, jolla on samat tiedot, mutta eri ulottuvuuksilla.
Tämä artikkeli havainnollistaa menetelmää käyttää laajennustoimintoa tensoreissa PyTorchissa.
Kuinka käyttää laajennustoimintoa PyTorchissa?
Jos haluat käyttää laajennustoimintoa PyTorchissa, tutustu alla oleviin vaiheisiin:
- Tuo PyTorch-kirjasto
- Luo haluttu tensori
- Näytä syötteen tensorin koko
- Laajenna tensoria käyttämällä ' laajentaa() ”-attribuutti
- Näytä laajennettu tensori ja sen koko
Vaihe 1: Tuo PyTorch-kirjasto
Tuo ensin ' taskulamppu ”-kirjasto käyttää laajennustoimintoa:
tuonti taskulamppu
Vaihe 2: Luo tensori
Luo sitten haluamasi tensori käyttämällä ' torch.tensor() ”-toiminto ja tulostaa sen elementit. Täällä luomme seuraavan ' Kymmeniä 'tensori:
Kymmeniä = taskulamppu. tensori ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
Tulosta ( Kymmeniä )
Alla olevassa lähdössä tensori on luotu onnistuneesti:
Vaihe 3: Näytä syötteen tensorin koko
Katso seuraavaksi yllä luodun ' Kymmeniä ' tensori käyttämällä ' koko() 'määrite:
Tulosta ( 'Tensorin koko:' , Kymmeniä. koko ( ) )Alla olevan lähdön mukaan tensorin koko on 3×1:
Vaihe 4: Laajenna Tensori
Käytä nyt ' laajentaa() ”-attribuutti suorittaa laajennusoperaation ja laajentaa tensorin uuteen ulottuvuuteen. Syötteenä on annettava tensori ja kokoluettelo. Tässä laajennamme tensoria 3×4-kokoon:
Exp_tens = Kymmeniä. laajentaa ( 3 , 4 )Vaihe 5: Näytä laajennettu tensori ja sen koko
Tulosta lopuksi laajennetun tensorin elementit ja sen koko:
Tulosta ( Exp_tens )Tulosta ( Exp_tens. koko ( ) )
Alla oleva tulos näyttää laajennetun tensorin ja sen koon eli 3 × 4. Tämä osoittaa, että laajennettu toiminto on suoritettu onnistuneesti:
Olemme selittäneet tehokkaan tavan käyttää laajennustoimintoa PyTorchissa
Huomautus : Pääset käyttämään Google Colab -muistikirjaamme tästä linkki .
Johtopäätös
Jos haluat käyttää laajennustoimintoa PyTorchissa, tuo ensin taskulamppukirjasto. Luo sitten haluttu tensori ja tarkastele sen elementtejä ja kokoa. Käytä seuraavaksi ' laajentaa() ”-attribuutti laajentaa syöttötensoria. Tulosta lopuksi laajennettu tensori ja katso sen kokoa. Tämä artikkeli on osoittanut menetelmän käyttää laajennustoimintoa tensoreissa PyTorchissa.