टेंसर को स्क्वीज़ करने के लिए, हम torch.squeeze() मेथड का उपयोग करते हैं। यह इनपुट टेंसर के सभी आयामों के साथ एक नया टेंसर लौटाता है लेकिन आकार 1 को हटा देता है। उदाहरण के लिए, यदि इनपुट टेंसर का आकार (M × 1 × N × 1 × P) है, तो स्क्वीज़ किए गए टेंसर का आकार (M × N × P) होगा।
टेंसर को अनस्क्वीज़ करने के लिए, हम torch.unsqueeze() मेथड का उपयोग करते हैं। यह विशिष्ट स्थिति पर आकार 1 का एक नया आयाम डालता है।
चरण
आवश्यक लाइब्रेरी इम्पोर्ट करें। निम्नलिखित सभी Python उदाहरणों में, आवश्यक Python लाइब्रेरी torch है। सुनिश्चित करें कि आपने इसे पहले ही इंस्टॉल कर लिया है।
एक टेंसर बनाएं और प्रिंट करें।
torch.squeeze(input) की गणना करें। यह आकार 1 को स्क्वीज़ (हटाता) है और input टेंसर के अन्य सभी आयामों के साथ एक टेंसर लौटाता है।
torch.unsqueeze(input, dim) की गणना करें। यह दिए गए dim पर आकार 1 का एक नया आयाम डालता है और टेंसर लौटाता है।
स्क्वीज़ और/या अनस्क्वीज़ किए गए टेंसर को प्रिंट करें।
उदाहरण 1
# टेंसर को स्क्वीज़ और अनस्क्वीज़ करने के लिए Python प्रोग्राम
# आवश्यक लाइब्रेरी इम्पोर्ट करें
import torch
# सभी एक का टेंसर बनाएं
T = torch.ones(2, 1, 2) # आकार 2x1x2
print("Original Tensor T:\n", T)
print("Size of T:", T.size())
# टेंसर के आयाम को स्क्वीज़ करें
squeezed_T = torch.squeeze(T) # अब आकार 2x2
print("Squeezed_T\n:", squeezed_T)
print("Size of Squeezed_T:", squeezed_T.size())
आउटपुट
Original Tensor T: tensor([[[1., 1.]],
[[1., 1.]]])
Size of T: torch.Size([2, 1, 2])
Squeezed_T: tensor([[1., 1.],
[1., 1.]])
Size of Squeezed_T: torch.Size([2, 2])
उदाहरण 2
# टेंसर को स्क्वीज़ और अनस्क्वीज़ करने के लिए Python प्रोग्राम
# आवश्यक लाइब्रेरी इम्पोर्ट करें
import torch
# एक टेंसर बनाएं
T = torch.Tensor([1, 2, 3]) # आकार 3
print("Original Tensor T:\n", T)
print("Size of T:", T.size())
# टेंसर को आयाम 0 या कॉलम dim में अनस्क्वीज़ करें
unsqueezed_T = torch.unsqueeze(T, dim=0) # अब आकार 1x3
print("Unsqueezed T\n:", unsqueezed_T)
print("Size of UnSqueezed T:", unsqueezed_T.size())
# टेंसर को आयाम 1 या रो dim में अनस्क्वीज़ करें
unsqueezed_T = torch.unsqueeze(T, dim=1) # अब आकार 3x1
print("Unsqueezed T\n:", unsqueezed_T)
print("Size of Unsqueezed T:", unsqueezed_T.size())
आउटपुट
Original Tensor T: tensor([1., 2., 3.])
Size of T: torch.Size([3])
Unsqueezed T: tensor([[1., 2., 3.]])
Size of UnSqueezed T: torch.Size([1, 3])
Unsqueezed T: tensor([[1.],
[2.],
[3.]])
Size of Unsqueezed T: torch.Size([3, 1])
