PyTorch में टेंसर के तत्वों को सॉर्ट करने के लिए, हम टॉर्च.सॉर्ट () विधि का उपयोग कर सकते हैं। यह विधि दो टेंसर लौटाती है। पहला टेंसर तत्वों के क्रमबद्ध मूल्यों के साथ एक टेंसर है और दूसरा टेंसर मूल टेंसर में तत्वों के सूचकांकों का एक टेंसर है। हम पंक्ति-वार और स्तंभ-वार 2D टेंसर की गणना कर सकते हैं।
कदम
-
आवश्यक पुस्तकालय आयात करें। निम्नलिखित सभी पायथन उदाहरणों में, आवश्यक पायथन पुस्तकालय है टॉर्च . सुनिश्चित करें कि आपने इसे पहले ही इंस्टॉल कर लिया है।
-
एक PyTorch टेंसर बनाएं और उसे प्रिंट करें।
-
ऊपर बनाए गए टेंसर के तत्वों को सॉर्ट करने के लिए, torch.sort(input, dim) की गणना करें . यह मान एक नए चर "v" . को निर्दिष्ट करें .यहां, इनपुट इनपुट टेंसर है और मंद वह आयाम है जिसके साथ तत्वों को क्रमबद्ध किया जाता है। तत्वों को पंक्ति-वार क्रमित करने के लिए, मंद को 1 के रूप में सेट किया गया है और तत्वों को स्तंभ-वार क्रमित करने के लिए मंद 0 के रूप में सेट है।
-
सॉर्ट किए गए मानों वाले टेंसर को v[0] . के रूप में एक्सेस किया जा सकता है और सॉर्ट किए गए तत्वों के सूचकांकों का टेंसर v[1] . के रूप में ।
-
सॉर्ट किए गए मानों के साथ टेंसर और सॉर्ट किए गए मानों के सूचकांक के साथ टेंसर प्रिंट करें।
उदाहरण 1
निम्नलिखित पायथन प्रोग्राम दिखाता है कि 1Dtensor के तत्वों को कैसे क्रमबद्ध किया जाए।
# Python program to sort elements of a tensor
# import necessary library
import torch
# Create a tensor
T = torch.Tensor([2.334,4.433,-4.33,-0.433,5, 4.443])
print("Original Tensor:\n", T)
# sort the tensor T
# it sorts the tensor in ascending order
v = torch.sort(T)
# print(v)
# print tensor of sorted value
print("Tensor with sorted value:\n", v[0])
# print indices of sorted value
print("Indices of sorted value:\n", v[1]) आउटपुट
Original Tensor: tensor([ 2.3340, 4.4330, -4.3300, -0.4330, 5.0000, 4.4430]) Tensor with sorted value: tensor([-4.3300, -0.4330, 2.3340, 4.4330, 4.4430, 5.0000]) Indices of sorted value: tensor([2, 3, 0, 1, 5, 4])
उदाहरण 2
निम्नलिखित पायथन प्रोग्राम दिखाता है कि 2Dtensor के तत्वों को कैसे क्रमबद्ध किया जाए।
# Python program to sort elements of a 2-D tensor
# import the library
import torch
# Create a 2-D tensor
T = torch.Tensor([[2,3,-32],
[43,4,-53],
[4,37,-4],
[3,-75,34]])
print("Original Tensor:\n", T)
# sort tensor T
# it sorts the tensor in ascending order
v = torch.sort(T)
# print(v)
# print tensor of sorted value
print("Tensor with sorted value:\n", v[0])
# print indices of sorted value
print("Indices of sorted value:\n", v[1])
print("Sort tensor Column-wise")
v = torch.sort(T, 0)
# print(v)
# print tensor of sorted value
print("Tensor with sorted value:\n", v[0])
# print indices of sorted value
print("Indices of sorted value:\n", v[1])
print("Sort tensor Row-wise")
v = torch.sort(T, 1)
# print(v)
# print tensor of sorted value
print("Tensor with sorted value:\n", v[0])
# print indices of sorted value
print("Indices of sorted value:\n", v[1]) आउटपुट
Original Tensor: tensor([[ 2., 3., -32.], [ 43., 4., -53.], [ 4., 37., -4.], [ 3., -75., 34.]]) Tensor with sorted value: tensor([[-32., 2., 3.], [-53., 4., 43.], [ -4., 4., 37.], [-75., 3., 34.]]) Indices of sorted value: tensor([[2, 0, 1], [2, 1, 0], [2, 0, 1], [1, 0, 2]]) Sort tensor Column-wise Tensor with sorted value: tensor([[ 2., -75., -53.], [ 3., 3., -32.], [ 4., 4., -4.], [ 43., 37., 34.]]) Indices of sorted value: tensor([[0, 3, 1], [3, 0, 0], [2, 1, 2], [1, 2, 3]]) Sort tensor Row-wise Tensor with sorted value: tensor([[-32., 2., 3.], [-53., 4., 43.], [ -4., 4., 37.], [-75., 3., 34.]]) Indices of sorted value: tensor([[2, 0, 1], [2, 1, 0], [2, 0, 1], [1, 0, 2]])