Kako dodati dimenzijo tenzorju v PyTorchu?

Ta članek bo ponazoril metodo dodajanja dimenzij Tensorju v PyTorchu.

Kako dodati dimenzijo specifičnemu tenzorju v PyTorchu?

Uporabniki lahko dodajajo dimenzije kateremu koli tenzorju, kot je 1D tenzor ali 2D tenzor v PyTorchu. Če želite dodati nove dimenzije tenzorjem na določenem položaju, si za boljše razumevanje oglejte naslednje primere:

• 1. primer: dodajte dimenzijo 1D tenzorju v PyTorchu
• 2. primer: dodajte dimenzijo 2D tenzorju v PyTorchu

1. primer: dodajte dimenzijo 1D tenzorju v PyTorchu

V tem primeru bomo ustvarili 1D tenzor in mu dodali dimenzijo na določenem položaju. Za praktično predstavitev sledite spodnjim korakom:

1. korak: Uvozite knjižnico
Najprej uvozite knjižnico torch:

2. korak: Ustvarite 1D tenzor
Nato ustvarite enodimenzionalni tenzor. Na primer, ustvarili smo naslednji tenzor in ga shranili v ' x ” spremenljivka:

3. korak: Oglejte si obliko tenzorja
Nato prikažite novo ustvarjeno obliko tenzorja, da si ogledate njene dimenzije:

Spodnji rezultat kaže, da je naš tenzor enodimenzionalen:

4. korak: dodajte dimenzijo 1D tenzorju
Zdaj pa uporabite » torch.unsqueeze(vhod, dim) ” za dodajanje dimenzije 1D tenzorju na določenem položaju. Na primer, dodajamo dimenzijo tenzorju pri indeksu 0:

tukaj,

• “ nove_desetice ” je spremenljivka, ki vključuje dodano dimenzijo.
• “ x ” je vhodni tenzor.
• “ dim=0 ” se uporablja za dodajanje dimenzije pri indeksu 0.

5. korak: Preverite izhod
Na koncu se prepričajte, ali je bila tenzorju dodana nova dimenzija ali ne:

V spodnjem rezultatu je mogoče opaziti, da je bila nova dimenzija dodana 1D tenzorju pri indeksu 0:

Poleg tega lahko uporabniki dodajo dimenzije tudi drugim položajem. Tukaj smo dodali dimenzijo v prvi indeks:

2. primer: dodajte dimenzijo 2D tenzorju v PyTorchu

Tukaj bomo ustvarili/naredili 2D tenzor in mu dodali dimenzijo na določenem položaju. Preizkusite podane korake za praktično izvedbo:

1. korak: uvozite knjižnico Torch
Najprej uvozite knjižnico torch:

2. korak: Ustvarite 2D tenzor
Nato ustvarite dvodimenzionalni tenzor. Na primer, ustvarili smo naslednji tenzor in ga shranili v ' x ” spremenljivka:

3. korak: Oglejte si obliko tenzorja
Po tem prikažite novo ustvarjeno obliko tenzorja, da si ogledate njene dimenzije:

Glede na spodnji rezultat je ta tenzor dvodimenzionalen:

4. korak: dodajte dimenzijo 2D tenzorju
Zdaj dodajte dimenzijo 2D tenzorju na določenem položaju z uporabo “torch.unsqueeze(input, dim)” funkcijo. Na primer, dodajamo dimenzijo tenzorju pri indeksu 0:

5. korak: Preverite izhod
Na koncu preverite, ali je bila nova dimenzija dodana 2D tenzorju ali ne:

Spodnji rezultat kaže, da je bila nova dimenzija uspešno dodana 2D tenzorju pri indeksu 0:

Opomba : Do našega zvezka Google Colab lahko dostopate tukaj povezava .

Na različnih primerih smo učinkovito razložili metodo dodajanja dimenzije tenzorju v PyTorchu.

Zaključek

Če želite dodati dimenzije tenzorju v PyTorchu, najprej uvozite knjižnico PyTorch. Nato ustvarite 1D ali 2D tenzor in si oglejte njegove dimenzije. Nato dodajte dimenzijo tenzorju na določenem položaju z uporabo ' torch.unsqueeze(vhod, dim) ”. Uporabniki morajo tej funkciji posredovati vhodni tenzor in želeni položaj indeksa kot parameter. Ta članek je ponazoril metodo dodajanja dimenzij tenzorju v PyTorchu.