Kako dodati dimenziju tenzoru u PyTorchu?

Ovaj će članak ilustrirati metodu dodavanja dimenzija tenzoru u PyTorchu.

Kako dodati dimenziju određenom tenzoru u PyTorchu?

Korisnici mogu dodati dimenzije bilo kojem tenzoru, kao što je 1D tenzor ili 2D tenzor u PyTorchu. Za dodavanje novih dimenzija tenzorima na određenoj poziciji, pogledajte sljedeće primjere radi boljeg razumijevanja:

• Primjer 1: Dodajte dimenziju 1D tenzoru u PyTorchu
• Primjer 2: Dodajte dimenziju 2D tenzoru u PyTorchu

Primjer 1: Dodajte dimenziju 1D tenzoru u PyTorchu

U ovom primjeru stvorit ćemo 1D tenzor i dodati mu dimenziju na određenoj poziciji. Slijedite dolje navedene korake za praktičnu demonstraciju:

Korak 1: Uvoz biblioteke
Prvo uvezite torch biblioteku:

Korak 2: Izradite 1D tenzor
Zatim stvorite jednodimenzionalni tenzor. Na primjer, stvorili smo sljedeći tenzor i pohranili ga u ' x ” varijabla:

Korak 3: Pregledajte oblik tenzora
Zatim prikažite novostvoreni oblik tenzora da vidite njegove dimenzije:

Donji rezultat pokazuje da je naš tenzor jednodimenzionalan:

Korak 4: Dodajte dimenziju 1D tenzoru
Sada upotrijebite ' torch.unsqueeze(unos, dim) ” za dodavanje dimenzije 1D tenzoru na određenoj poziciji. Na primjer, dodajemo dimenziju tenzoru na 0 indeksu:

Ovdje,

• “ nove_desetice ” je varijabla koja uključuje dodanu dimenziju.
• “ x ” je ulazni tenzor.
• “ dim=0 ” koristi se za dodavanje dimenzije na 0 indeks.

Korak 5: Provjerite izlaz
Na kraju, provjerite je li nova dimenzija dodana tenzoru ili ne:

U donjem izlazu može se uočiti da je nova dimenzija dodana 1D tenzoru na indeksu 0:

Štoviše, korisnici također mogu dodati dimenzije drugim pozicijama. Ovdje smo dodali dimenziju u prvi indeks:

Primjer 2: Dodajte dimenziju 2D tenzoru u PyTorchu

Ovdje ćemo kreirati/napraviti 2D tenzor i dodati mu dimenziju na određenoj poziciji. Isprobajte navedene korake za praktičnu primjenu:

Korak 1: Uvezite Torch biblioteku
Prvo uvezite torch biblioteku:

Korak 2: Napravite 2D tenzor
Zatim stvorite dvodimenzionalni tenzor. Na primjer, stvorili smo sljedeći tenzor i pohranili ga u ' x ” varijabla:

Korak 3: Pregledajte oblik tenzora
Nakon toga, prikažite novostvoreni oblik tenzora da vidite njegove dimenzije:

Prema rezultatu ispod, ovaj tenzor je dvodimenzionalan:

Korak 4: Dodajte dimenziju 2D tenzoru
Sada dodajte dimenziju 2D tenzoru na određenoj poziciji pomoću “torch.unsqueeze(input, dim)” funkcija. Na primjer, dodajemo dimenziju tenzoru na 0 indeksu:

Korak 5: Provjerite izlaz
Na kraju, provjerite je li nova dimenzija dodana 2D tenzoru ili ne:

Donji rezultat pokazuje da je nova dimenzija uspješno dodana 2D tenzoru na indeksu 0:

Bilješka : Našoj Google Colab bilježnici možete pristupiti ovdje veza .

Učinkovito smo objasnili metodu dodavanja dimenzije tenzoru u PyTorchu koristeći različite primjere.

Zaključak

Za dodavanje dimenzija tenzoru u PyTorchu, prvo uvezite biblioteku PyTorcha. Zatim izradite 1D ili 2D tenzor i pogledajte njegove dimenzije. Nakon toga, dodajte dimenziju tenzoru na određenoj poziciji pomoću ' torch.unsqueeze(unos, dim) ” funkcija. Korisnici trebaju proslijediti ulazni tenzor i željenu poziciju indeksa kao parametar ovoj funkciji. Ovaj članak je pokazao primjer metode dodavanja dimenzija tenzoru u PyTorchu.