Paano Magdagdag ng Dimensyon sa isang Tensor sa PyTorch?

Ipapakita ng artikulong ito ang paraan ng pagdaragdag ng mga dimensyon sa isang Tensor sa PyTorch.

Paano Magdagdag ng Dimensyon sa Tiyak na Tensor sa PyTorch?

Maaaring magdagdag ng mga dimensyon ang mga user sa anumang Tensor, gaya ng 1D tensor o 2D tensor sa PyTorch. Upang magdagdag ng mga bagong dimensyon sa mga tensor sa isang partikular na posisyon, tingnan ang mga sumusunod na halimbawa para sa isang mas mahusay na pag-unawa:

• Halimbawa 1: Magdagdag ng Dimensyon sa isang 1D Tensor sa PyTorch
• Halimbawa 2: Magdagdag ng Dimensyon sa isang 2D Tensor sa PyTorch

Halimbawa 1: Magdagdag ng Dimensyon sa isang 1D Tensor sa PyTorch

Sa halimbawang ito, gagawa kami ng 1D tensor at magdagdag ng dimensyon dito sa isang partikular na posisyon. Sundin ang mga nakalistang hakbang sa ibaba para sa isang praktikal na pagpapakita:

Hakbang 1: Mag-import ng Library
Una, i-import ang torch library:

Hakbang 2: Lumikha ng 1D Tensor
Pagkatapos, lumikha ng one-dimensional tensor. Halimbawa, nilikha namin ang sumusunod na tensor at inimbak ito sa ' x ” variable:

Hakbang 3: Tingnan ang Tensor Shape
Susunod, ipakita ang bagong likhang hugis ng tensor upang tingnan ang mga sukat nito:

Ang output sa ibaba ay nagpapahiwatig na ang aming tensor ay one-dimensional:

Hakbang 4: Magdagdag ng Dimensyon sa 1D Tensor
Ngayon, gamitin ang ' torch.unsqueeze(input, dim) ” function upang idagdag ang dimensyon sa 1D tensor sa partikular na posisyon. Halimbawa, idinaragdag namin ang dimensyon sa tensor sa 0 index:

dito,

• “ bagong_sampu ” ay ang variable na kasama ang idinagdag na dimensyon.
• “ x ” ay ang input tensor.
• “ madilim=0 Ang ” ay ginagamit upang magdagdag ng dimensyon sa 0 index.

Hakbang 5: I-verify ang Output
Panghuli, tiyaking may naidagdag na bagong dimensyon sa tensor o hindi:

Sa ibabang output, mapapansin na ang bagong dimensyon ay naidagdag sa 1D tensor sa 0 index:

Bukod dito, ang mga gumagamit ay maaari ring magdagdag ng mga sukat sa iba pang mga posisyon. Dito, idinagdag namin ang dimensyon sa unang index:

Halimbawa 2: Magdagdag ng Dimensyon sa isang 2D Tensor sa PyTorch

Dito, gagawa/gagawa kami ng 2D tensor at magdagdag ng dimensyon dito sa partikular na posisyon. Subukan ang ibinigay na mga hakbang para sa praktikal na pagpapatupad:

Hakbang 1: Mag-import ng Torch Library
Una, i-import ang torch library:

Hakbang 2: Gumawa ng 2D Tensor
Pagkatapos, lumikha ng dalawang-dimensional na tensor. Halimbawa, nilikha namin ang sumusunod na tensor at inimbak ito sa ' x ” variable:

Hakbang 3: Tingnan ang Tensor Shape
Pagkatapos nito, ipakita ang bagong likhang hugis ng tensor upang tingnan ang mga sukat nito:

Ayon sa output sa ibaba, ang tensor na ito ay dalawang-dimensional:

Hakbang 4: Magdagdag ng Dimensyon sa 2D Tensor
Ngayon, idagdag ang dimensyon sa 2D tensor sa partikular na posisyon gamit ang “torch.unsqueeze(input, dim)” function. Halimbawa, idinaragdag namin ang dimensyon sa tensor sa 0 index:

Hakbang 5: I-verify ang Output
Panghuli, i-verify kung ang bagong dimensyon ay naidagdag sa 2D tensor o hindi:

Ang output sa ibaba ay nagpapahiwatig na ang bagong dimensyon ay matagumpay na naidagdag sa 2D tensor sa 0 index:

Tandaan : Maa-access mo ang aming Google Colab Notebook dito link .

Mahusay naming naipaliwanag ang paraan ng pagdaragdag ng dimensyon sa isang tensor sa PyTorch gamit ang iba't ibang halimbawa.

Konklusyon

Upang magdagdag ng mga dimensyon sa isang tensor sa PyTorch, una, i-import ang library ng PyTorch. Pagkatapos, gumawa ng 1D o 2D tensor at tingnan ang mga dimensyon nito. Pagkatapos nito, idagdag ang dimensyon sa isang tensor sa partikular na posisyon sa pamamagitan ng paggamit ng “ torch.unsqueeze(input, dim) ” function. Kailangang ipasa ng mga user ang input tensor at ninanais na posisyon ng index bilang parameter sa function na ito. Inihalimbawa ng artikulong ito ang paraan ng pagdaragdag ng mga dimensyon sa isang tensor sa PyTorch.