কিভাবে PyTorch একটি টেনসর চেপে এবং unsqueeze?

একটি টেনসর চাপতে, আমরা torch.squeeze() ব্যবহার করি পদ্ধতি এটি ইনপুট টেনসরের সমস্ত মাত্রা সহ একটি নতুন টেনসর ফেরত দেয় কিন্তু আকার 1 সরিয়ে দেয়। উদাহরণস্বরূপ, যদি ইনপুট টেনসরের আকৃতি হয় (M ☓ 1 ☓ N ☓ 1 ☓ P), তাহলে চাপা টেনসরটির আকৃতি থাকবে ( M ☓ M ☓ P).

একটি টেনসর বের করতে, আমরা torch.unsqueeze() ব্যবহার করি পদ্ধতি এটি নির্দিষ্ট অবস্থানে ঢোকানো সাইজ 1 এর একটি নতুন টেনসর মাত্রা প্রদান করে।

পদক্ষেপ

• প্রয়োজনীয় লাইব্রেরি আমদানি করুন। নিম্নলিখিত সমস্ত পাইথন উদাহরণে, প্রয়োজনীয় পাইথন লাইব্রেরি হল টর্চ . নিশ্চিত করুন যে আপনি এটি ইতিমধ্যেই ইনস্টল করেছেন৷

  ৷

• একটি টেনসর তৈরি করুন এবং এটি মুদ্রণ করুন৷

  ৷

• torch.squeeze(ইনপুট) গণনা করুন . এটি সাইজ 1 চেপে (মুছে দেয়) এবং ইনপুট এর অন্যান্য সমস্ত মাত্রা সহ একটি টেনসর প্রদান করে টেনসর।

• কম্পিউট করুন torch.unsqueeze(ইনপুট, ম্লান) . এটি প্রদত্ত ম্লান এ সাইজ 1 এর একটি নতুন মাত্রা সন্নিবেশ করায় এবং টেনসর প্রদান করে৷

• স্কুইজড এবং/অথবা আনস্কুইজড টেনসর প্রিন্ট করুন।

উদাহরণ 1

# Python program to squeeze and unsqueeze a tensor
# import necessary library
import torch

# Create a tensor of all one
T = torch.ones(2,1,2) # size 2x1x2
print("Original Tensor T:\n", T )
print("Size of T:", T.size())

# Squeeze the dimension of the tensor
squeezed_T = torch.squeeze(T) # now size 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 program to squeeze and unsqueeze a tensor
# import necessary library
import torch

# create a tensor
T = torch.Tensor([1,2,3]) # size 3
print("Original Tensor T:\n", T )
print("Size of T:", T.size())

# Squeeze the tensor in dimension o or column dim
unsqueezed_T = torch.unsqueeze(T, dim = 0) # now size 1x3
print("Unsqueezed T\n:", unsqueezed_T )
print("Size of UnSqueezed T:", unsqueezed_T.size())

# Squeeze the tensor in dimension 1 or row dim
unsqueezed_T = torch.unsqueeze(T, dim = 1) # now size 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])