Kā PyTorch tensoram pievienot izmēru?

Šajā rakstā ir parādīts piemērs, kā PyTorch tensoram pievienot izmērus.

Kā pievienot izmēru konkrētajam tensoram programmā PyTorch?

Lietotāji var pievienot izmērus jebkuram Tensoram, piemēram, 1D tensoram vai 2D tensoram PyTorch. Lai pievienotu jaunas dimensijas tenzoriem noteiktā pozīcijā, labākai izpratnei skatiet tālāk norādītos piemērus.

• 1. piemērs: pievienojiet dimensiju 1D tensoram programmā PyTorch
• 2. piemērs: pievienojiet dimensiju 2D tensoram programmā PyTorch

1. piemērs: pievienojiet dimensiju 1D tensoram programmā PyTorch

Šajā piemērā mēs izveidosim 1D tensoru un pievienosim tam dimensiju noteiktā vietā. Lai iegūtu praktisku demonstrāciju, izpildiet tālāk norādītās darbības.

1. darbība. Importējiet bibliotēku
Vispirms importējiet lāpas bibliotēku:

2. darbība. Izveidojiet 1D tensoru
Pēc tam izveidojiet viendimensijas tensoru. Piemēram, mēs esam izveidojuši šādu tensoru un saglabājuši to mapē “ x ” mainīgais:

3. darbība: skatiet tenzora formu
Pēc tam parādiet jaunizveidoto tenzora formu, lai skatītu tās izmērus:

Zemāk redzamā izvade norāda, ka mūsu tensors ir viendimensionāls:

4. darbība: pievienojiet izmēru 1D tensoram
Tagad izmantojiet ' torch.unsqueeze (ievade, aptumšot) ” funkciju, lai pievienotu dimensiju 1D tensoram konkrētajā vietā. Piemēram, mēs pievienojam dimensiju tensoram ar indeksu 0:

Šeit,

• ' jauni_desmitie ” ir mainīgais, kas ietver pievienoto dimensiju.
• ' x ” ir ievades tensors.
• ' dim=0 ” tiek izmantots, lai pievienotu dimensiju 0 indeksam.

5. darbība: pārbaudiet izvadi
Visbeidzot, pārliecinieties, vai tensoram ir pievienota jauna dimensija:

Zemāk esošajā izvadā var novērot, ka jaunā dimensija ir pievienota 1D tensoram pie 0 indeksa:

Turklāt lietotāji var pievienot izmērus arī citām pozīcijām. Šeit mēs esam pievienojuši dimensiju pirmajā rādītājā:

2. piemērs: pievienojiet dimensiju 2D tensoram programmā PyTorch

Šeit mēs izveidosim/izveidosim 2D tensoru un pievienosim tam dimensiju konkrētajā vietā. Izmēģiniet norādītās darbības praktiskai īstenošanai:

1. darbība. Importējiet Torch bibliotēku
Vispirms importējiet lāpas bibliotēku:

2. darbība: izveidojiet 2D tensoru
Pēc tam izveidojiet divdimensiju tensoru. Piemēram, mēs esam izveidojuši šādu tensoru un saglabājuši to mapē “ x ” mainīgais:

3. darbība: skatiet tenzora formu
Pēc tam parādiet jaunizveidoto tenzora formu, lai skatītu tās izmērus:

Saskaņā ar zemāk redzamo izvadi šis tensors ir divdimensiju:

4. darbība: pievienojiet izmēru 2D tensoram
Tagad pievienojiet izmēru 2D tensoram konkrētajā vietā, izmantojot “torch.unsqueeze(input, dim)” funkciju. Piemēram, mēs pievienojam dimensiju tensoram ar indeksu 0:

5. darbība: pārbaudiet izvadi
Visbeidzot pārbaudiet, vai 2D tensoram ir pievienota jaunā dimensija:

Tālāk redzamā izvade norāda, ka jaunā dimensija ir veiksmīgi pievienota 2D tensoram ar indeksu 0:

Piezīme : varat piekļūt mūsu Google Colab piezīmju grāmatiņai šeit saite .

Mēs esam efektīvi izskaidrojuši metodi, kā PyTorch tensoram pievienot dimensiju, izmantojot dažādus piemērus.

Secinājums

Lai PyTorch tensoram pievienotu izmērus, vispirms importējiet PyTorch bibliotēku. Pēc tam izveidojiet 1D vai 2D tensoru un skatiet tā izmērus. Pēc tam pievienojiet dimensiju tensoram konkrētajā vietā, izmantojot ' torch.unsqueeze (ievade, aptumšot) ” funkcija. Lietotājiem kā parametrs šai funkcijai ir jānodod ievades tensors un vēlamā indeksa pozīcija. Šajā rakstā ir parādīts piemērs, kā PyTorch tensoram pievienot izmērus.