Kaip naudoti išplėtimo operaciją „PyTorch“?

PyTorch yra mašininio mokymosi biblioteka, leidžianti vartotojams dirbti su tensoriais. Tenzoriai yra esminės duomenų struktūros, kurios gali turėti įvairių tipų duomenis ir formas. Kartais vartotojai gali norėti atlikti tenzorių išplėtimo operaciją, kad padidintų jų dydį. Išplėtimo operacija leidžia vartotojams pakartoti tenzorių pagal nurodytus matmenis. „PyTorch“ pateikia atributą „expand()“, kuris kaip įvestį naudoja tenzorių ir dydžių sąrašą. Jis grąžina naują tenzorių, turintį tuos pačius duomenis, bet skirtingų matmenų.

Šiame straipsnyje bus parodytas metodas, kaip naudoti „PyTorch“ tenzorių išplėtimo operaciją.

Kaip naudoti išplėtimo operaciją „PyTorch“?

Norėdami naudoti išplėtimo operaciją PyTorch, patikrinkite toliau pateiktus veiksmus:

• Importuokite PyTorch biblioteką
• Sukurkite norimą tenzorių
• Peržiūrėkite įvesties tenzoriaus dydį
• Išplėskite tenzorių naudodami ' išplėsti () “ atributas
• Rodyti išplėstinį tenzorių ir jo dydį

1 veiksmas: importuokite PyTorch biblioteką

Pirmiausia importuokite „ deglas “ biblioteka, kad galėtumėte naudoti išplėtimo operaciją:

2 veiksmas: sukurkite tensorį

Tada sukurkite norimą tenzorių naudodami „ torch.tensor() “ funkciją ir atspausdinti jos elementus. Čia mes kuriame šiuos ' Dešimtys “ tenzorius:

Žemiau pateiktame išvestyje tenzorius buvo sėkmingai sukurtas:

3 veiksmas: peržiūrėkite įvesties tenzoriaus dydį

Tada peržiūrėkite aukščiau sukurto ' Dešimtys “ tenzorius naudojant “ dydis () “ atributas:

Pagal žemiau pateiktą išvestį tenzoriaus dydis yra 3 × 1:

4 veiksmas: išplėskite Tensor

Dabar naudokite „ išplėsti () “ atributas, kad atliktumėte išplėtimo operaciją ir išplėstumėte tenzorių į naują dimensiją. Kaip įvestį reikia pateikti tenzorių ir dydžių sąrašą. Čia mes išplečiame tenzorių iki 3 × 4 dydžio:

5 veiksmas: parodykite išplėstinį tenzorių ir jo dydį

Galiausiai atspausdinkite išplėstinio tenzoriaus elementus ir jo dydį:

Žemiau esančioje išvestyje parodytas išplėstinis tenzorius ir jo dydis, ty 3 × 4. Tai rodo, kad išplėstinė operacija buvo sėkmingai atlikta:

Mes paaiškinome efektyvų metodą, kaip naudoti išplėtimo operaciją „PyTorch“.

Pastaba : „Google Colab“ bloknotą galite pasiekti šiuo adresu nuoroda .

Išvada

Norėdami naudoti išplėtimo operaciją PyTorch, pirmiausia importuokite žibintuvėlių biblioteką. Tada sukurkite norimą tenzorių ir peržiūrėkite jo elementus bei dydį. Tada naudokite „ išplėsti () “ atributas įvesties tenzoriui išplėsti. Galiausiai atspausdinkite išplėstinį tenzorių ir peržiūrėkite jo dydį. Šiame straipsnyje parodytas metodas, kaip naudoti „PyTorch“ tenzorių išplėtimo operaciją.