Com utilitzar l'operació d'expandir a PyTorch?

Com Utilitzar L Operacio D Expandir A Pytorch



PyTorch és una biblioteca d'aprenentatge automàtic que permet als usuaris treballar amb tensors. Els tensors són estructures de dades essencials que poden tenir diferents tipus de dades i formes. De vegades, és possible que els usuaris vulguin realitzar l'operació d'expansió dels tensors per ampliar la seva mida. L'operació d'expandir permet als usuaris repetir un tensor al llarg de les dimensions especificades. PyTorch proporciona l'atribut 'expand()' que pren un tensor i una llista de mides com a entrades. Retorna un nou tensor que té les mateixes dades però amb dimensions diferents.

Aquest article il·lustrarà el mètode per utilitzar l'operació d'expandir en tensors a PyTorch.







Com utilitzar l'operació d'expandir a PyTorch?

Per utilitzar l'operació d'expansió a PyTorch, consulteu els passos que s'indiquen a continuació:



Pas 1: importa la biblioteca PyTorch

Primer, importeu el ' torxa ” biblioteca per utilitzar l'operació d'expandir:



importar torxa

Pas 2: creeu un tensor

A continuació, creeu el tensor desitjat amb el ' torch.tensor() ” funcionen i imprimeixen els seus elements. Aquí, estem creant el següent ' Desenes ” tensor:





Desenes = torxa. tensor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )

imprimir ( Desenes )

A la sortida següent, el tensor s'ha creat correctament:



Pas 3: visualitzeu la mida del tensor d'entrada

A continuació, visualitzeu la mida de l'element ' Desenes ' tensor utilitzant el ' mida () ” atribut:

imprimir ( 'Mida del tensor:' , Desenes. mida ( ) )

Segons la sortida següent, la mida del tensor és 3×1:

Pas 4: expandeix Tensor

Ara, utilitzeu el ' expandir () ” per realitzar l'operació d'expandir i ampliar el tensor a una nova dimensió. Cal proporcionar el tensor i la llista de mides com a entrada. Aquí, estem ampliant el tensor a mida 3×4:

Exp_tens = Desenes. expandir ( 3 , 4 )

Pas 5: Mostra el tensor ampliat i la seva mida

Finalment, imprimiu els elements del tensor expandit i la seva mida:

imprimir ( Exp_tens )

imprimir ( Exp_tens. mida ( ) )

La sortida següent mostra el tensor expandit i la seva mida, és a dir, 3×4. Això indica que l'operació ampliada s'ha realitzat correctament:

Hem explicat el mètode eficient per utilitzar l'operació d'expansió a PyTorch

Nota : Podeu accedir al nostre quadern Google Colab aquí enllaç .

Conclusió

Per utilitzar l'operació d'expansió a PyTorch, primer, importeu la biblioteca de la torxa. A continuació, creeu el tensor desitjat i visualitzeu-ne els elements i la mida. A continuació, utilitzeu el ' expandir () ” atribut per ampliar el tensor d'entrada. Finalment, imprimiu el tensor expandit i visualitzeu-ne la mida. Aquest article ha demostrat el mètode per utilitzar l'operació d'expandir en tensors a PyTorch.