यह ब्लॉग PyTorch में 'torch.argmax()' विधि का उपयोग करने की विधि का उदाहरण देगा।
PyTorch में 'torch.argmax()' विधि का उपयोग कैसे करें?
'Torch.argmax()' विधि किसी भी 1D या 2D टेंसर को इनपुट के रूप में लेती है और एक टेंसर लौटाती है जिसमें दिए गए आयाम के साथ अधिकतम मानों के सूचकांक/सूचकांक होते हैं।
'Torch.argmax()' विधि का सिंटैक्स नीचे दिया गया है:
मशाल. argmax ( < इनपुट_टेंसर > )
PyTorch में इस पद्धति का उपयोग करने के लिए, बेहतर समझ के लिए निम्नलिखित उदाहरण देखें:
उदाहरण 1: 1डी टेंसर के साथ 'torch.argmax()' विधि का उपयोग करें
पहले उदाहरण में, हम एक 1D टेंसर बनाएंगे और उसके साथ 'torch.argmax()' विधि का उपयोग करेंगे। आइए नीचे दी गई चरण-दर-चरण प्रक्रिया का पालन करें:
चरण 1: PyTorch लाइब्रेरी आयात करें
सबसे पहले, 'आयात करें' मशाल 'टॉर्च.आर्गमैक्स()' विधि का उपयोग करने के लिए लाइब्रेरी:
आयात मशालचरण 2: 1D टेंसर बनाएं
फिर, एक 1डी टेंसर बनाएं और उसके तत्वों को प्रिंट करें। यहां, हम निम्नलिखित बना रहे हैं ' दहाई 1 ' का उपयोग करके सूची से टेंसर टॉर्च.टेंसर() ' समारोह:
दहाई 1 = मशाल. टेन्सर ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )
छपाई ( दहाई 1 )
इसने एक 1D टेंसर बनाया है जैसा कि नीचे देखा गया है:
चरण 3: अधिकतम मूल्य के सूचकांक खोजें
अब, 'का उपयोग करें टॉर्च.आर्गमैक्स() 'में अधिकतम मूल्य के सूचकांक/सूचकांक खोजने के लिए फ़ंक्शन' दहाई 1 ' टेंसर:
T1_ind = मशाल. argmax ( दहाई 1 )चरण 4: अधिकतम मूल्य का सूचकांक प्रिंट करें
अंत में, इनपुट टेंसर में अधिकतम मान का सूचकांक प्रदर्शित करें:
छपाई ( 'सूचकांक:' , T1_ind )नीचे दिया गया आउटपुट 'में अधिकतम मूल्य का सूचकांक दिखाता है' दहाई 1 'टेंसर यानी, 4. इसका मतलब है कि टेन्सर का उच्चतम मूल्य चौथे सूचकांक पर है जो ' है 9 ”:
उदाहरण 2: 2डी टेंसर के साथ 'torch.argmax()' विधि का उपयोग करें
दूसरे उदाहरण में, हम एक 2D टेंसर बनाएंगे और उसके साथ 'torch.argmax()' विधि का उपयोग करेंगे। आइए दिए गए चरणों का पालन करें:
चरण 1: PyTorch लाइब्रेरी आयात करें
सबसे पहले, 'आयात करें' मशाल 'टॉर्च.आर्गमैक्स()' विधि का उपयोग करने के लिए लाइब्रेरी:
आयात मशालचरण 2: 2डी टेंसर बनाएं
फिर, 'का उपयोग करें टॉर्च.टेंसर() 2डी टेंसर बनाने और उसके तत्वों को प्रिंट करने का कार्य। यहां, हम निम्नलिखित बना रहे हैं ' दहाई2 '2डी टेंसर:
दहाई2 = मशाल. टेन्सर ( [ [ 4 , 1 , - 7 ] , [ पंद्रह , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )छपाई ( दहाई2 )
इसने एक 2D टेंसर बनाया है जैसा कि नीचे देखा गया है:
चरण 3: अधिकतम मूल्य के सूचकांक खोजें
अब, 'में अधिकतम मूल्य का सूचकांक खोजें दहाई2 'का उपयोग करके टेंसर टॉर्च.आर्गमैक्स() ' समारोह:
T2_ind = मशाल. argmax ( दहाई2 )चरण 4: अधिकतम मूल्य का सूचकांक प्रिंट करें
अंत में, इनपुट टेंसर में अधिकतम मान का सूचकांक प्रदर्शित करें:
छपाई ( 'सूचकांक:' , T2_ind )नीचे दिए गए आउटपुट के अनुसार, 'में अधिकतम मूल्य का सूचकांक' दहाई2 'टेंसर' 3' है। इसका मतलब है कि टेंसर का उच्चतम मान तीसरे सूचकांक पर है जो ' पंद्रह ”:
चरण 5: कॉलमों के साथ अधिकतम मूल्य के सूचकांक खोजें
इसके अलावा, उपयोगकर्ता टेंसर के प्रत्येक कॉलम के साथ अधिकतम मानों के सूचकांक/सूचकांक भी पा सकते हैं। उदाहरण के लिए, हम 'का उपयोग कर सकते हैं मंद=0 'Torch.argmax()' फ़ंक्शन के साथ तर्क। यह 'में कॉलम के साथ अधिकतम मानों के सूचकांक ढूंढता है दहाई2 'टेंसर और फिर उन सूचकांकों को प्रिंट करता है:
col_index = मशाल. argmax ( दहाई2 , धुंधला = 0 )छपाई ( 'स्तंभों में सूचकांक:' , col_index )
नीचे दिया गया आउटपुट टेंसर के प्रत्येक कॉलम के साथ अधिकतम मानों के सूचकांक दिखाता है:
चरण 6: पंक्तियों के साथ अधिकतम मूल्य के सूचकांक खोजें
इसी प्रकार, उपयोगकर्ता टेंसर की प्रत्येक पंक्ति के साथ अधिकतम मानों के सूचकांक/सूचकांक भी पा सकते हैं। उदाहरण के लिए, ' का उपयोग करें मंद=1 'Tens2' टेंसर में पंक्तियों के साथ अधिकतम मानों के सूचकांकों को खोजने के लिए 'torch.argmax()' फ़ंक्शन के साथ तर्क करें और फिर उन सूचकांकों को प्रिंट करें:
row_index = मशाल. argmax ( दहाई2 , धुंधला = 1 )छपाई ( 'पंक्तियों में सूचकांक:' , row_index )
'Tens2' टेंसर की प्रत्येक पंक्ति के साथ अधिकतम मान के सूचकांक नीचे देखे जा सकते हैं:
हमने PyTorch में 'torch.argmax()' विधि का उपयोग करने की विधि को कुशलतापूर्वक समझाया है।
टिप्पणी : आप यहां हमारे Google Colab नोटबुक तक पहुंच सकते हैं जोड़ना .
निष्कर्ष
PyTorch में 'torch.argmax()' विधि का उपयोग करने के लिए, सबसे पहले, 'आयात करें' मशाल ' पुस्तकालय। फिर, वांछित 1D या 2D टेंसर बनाएं और उसके तत्वों को देखें। अगला, 'का उपयोग करें टॉर्च.आर्गमैक्स() 'टेंसर में अधिकतम मानों के सूचकांकों/सूचकांकों को खोजने/गणना करने की विधि। इसके अलावा, उपयोगकर्ता '' का उपयोग करके टेंसर में प्रत्येक पंक्ति या स्तंभ के साथ अधिकतम मूल्य के सूचकांक भी पा सकते हैं धुंधला ' तर्क। अंत में, इनपुट टेंसर में अधिकतम मान का सूचकांक प्रदर्शित करें। इस ब्लॉग ने PyTorch में 'torch.argmax()' पद्धति का उपयोग करने की विधि का उदाहरण दिया है।