PyTorch è una libreria di apprendimento automatico che consente agli utenti di lavorare con tensori. I tensori sono strutture dati essenziali che possono avere diversi tipi di dati e forme. A volte, gli utenti potrebbero voler eseguire l'operazione di espansione sui tensori per estenderne le dimensioni. L'operazione di espansione consente agli utenti di ripetere un tensore lungo dimensioni specificate. PyTorch fornisce l'attributo 'expand()' che accetta un tensore e un elenco di dimensioni come input. Restituisce un nuovo tensore che ha gli stessi dati ma con dimensioni diverse.
Questo articolo illustrerà il metodo per utilizzare l'operazione di espansione sui tensori in PyTorch.
Come utilizzare l'operazione di espansione in PyTorch?
Per utilizzare l'operazione di espansione in PyTorch, controlla i passaggi forniti di seguito:
- Importa la libreria PyTorch
- Crea il tensore desiderato
- Visualizza la dimensione del tensore di input
- Espandi il tensore usando il pulsante “ espandere() attributo
- Visualizza il tensore espanso e le sue dimensioni
Passaggio 1: importa la libreria PyTorch
Innanzitutto, importa il ' torcia ' libreria per utilizzare l'operazione di espansione:
importare torcia
Passaggio 2: creare un tensore
Quindi, crea il tensore desiderato utilizzando il comando ' torcia.tensore() 'funziona e stampa i suoi elementi. Qui, stiamo creando quanto segue “ Decine ' tensore:
Decine = torcia. tensore ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
stampa ( Decine )
Nell'output seguente, il tensore è stato creato con successo:
Passaggio 3: visualizzare la dimensione del tensore di input
Successivamente, visualizza la dimensione del 'creato sopra' Decine ' tensore utilizzando il ' misurare() attributo:
stampa ( 'Dimensione del tensore:' , Decine. misurare ( ) )Secondo l'output seguente, la dimensione del tensore è 3×1:
Passaggio 4: espandere il tensore
Ora usa il ' espandere() ” per eseguire l'operazione di espansione ed estendere il tensore a una nuova dimensione. È necessario fornire il tensore e l'elenco delle dimensioni come input. Qui stiamo espandendo il tensore alla dimensione 3×4:
Esp_decine = Decine. espandere ( 3 , 4 )Passaggio 5: Visualizza il tensore espanso e le sue dimensioni
Infine, stampa gli elementi del tensore espanso e la sua dimensione:
stampa ( Esp_decine )stampa ( Esp_decine. misurare ( ) )
L'output seguente mostra il tensore espanso e le sue dimensioni, ovvero 3×4. Ciò indica che l'operazione espansa è stata eseguita correttamente:
Abbiamo spiegato il metodo efficiente per utilizzare l'operazione di espansione in PyTorch
Nota : puoi accedere al nostro Google Colab Notebook da qui collegamento .
Conclusione
Per utilizzare l'operazione di espansione in PyTorch, importa innanzitutto la libreria della torcia. Quindi, crea il tensore desiderato e visualizzane gli elementi e le dimensioni. Successivamente, utilizzare il ' espandere() ” per espandere il tensore di input. Infine, stampa il tensore espanso e visualizzane le dimensioni. Questo articolo ha dimostrato il metodo per utilizzare l'operazione di espansione sui tensori in PyTorch.