ستوضح هذه المدونة طريقة استخدام طريقة 'torch.argmax()' في PyTorch.
كيفية استخدام طريقة 'torch.argmax ()' في PyTorch؟
تأخذ طريقة 'torch.argmax()' أي موتر أحادي الأبعاد أو ثنائي الأبعاد كمدخل وتعيد موترًا يحتوي على مؤشرات/فهرس القيم القصوى على طول البعد المحدد.
بناء جملة طريقة 'torch.argmax ()' موضح أدناه:
شعلة. com.argmax ( < input_tensor > )
لاستخدام هذه الطريقة في PyTorch، راجع الأمثلة التالية لفهم أفضل:
مثال 1: استخدم طريقة 'torch.argmax()' مع موتر 1D
في المثال الأول، سنقوم بإنشاء موتر أحادي الأبعاد واستخدام طريقة 'torch.argmax()' معه. دعنا نتبع الإجراء التالي خطوة بخطوة:
الخطوة 1: استيراد مكتبة PyTorch
أولاً، قم باستيراد ' شعلة مكتبة لاستخدام طريقة 'torch.argmax ()':
يستورد شعلةالخطوة 2: إنشاء موتر 1D
ثم قم بإنشاء موتر أحادي الأبعاد وطباعة عناصره. وهنا نقوم بإنشاء ما يلي ' عشرات1 'الموتر من قائمة باستخدام' الشعلة.الموتر() ' وظيفة:
عشرات1 = شعلة. الموتر ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )
مطبعة ( عشرات1 )
أدى هذا إلى إنشاء موتر 1D كما هو موضح أدناه:
الخطوة 3: ابحث عن مؤشرات القيمة القصوى
الآن استخدم ' الشعلة.argmax() وظيفة 'للعثور على الفهرس/المؤشرات ذات القيمة القصوى في' عشرات1 'الموتر:
T1_ind = شعلة. com.argmax ( عشرات1 )الخطوة 4: طباعة فهرس القيمة القصوى
وأخيرًا، قم بعرض فهرس القيمة القصوى في موتر الإدخال:
مطبعة ( 'المؤشرات:' , T1_ind )يُظهر الإخراج أدناه مؤشر القيمة القصوى في ' عشرات1 'الموتر أي 4. وهذا يعني أن أعلى قيمة للموتر تقع عند المؤشر الرابع وهو' 9 ':
مثال 2: استخدم طريقة 'torch.argmax()' مع Tensor ثنائي الأبعاد
في المثال الثاني، سنقوم بإنشاء موتر ثنائي الأبعاد واستخدام طريقة “torch.argmax()” معه. دعنا نتبع الخطوات المقدمة:
الخطوة 1: استيراد مكتبة PyTorch
أولاً، قم باستيراد ' شعلة مكتبة لاستخدام طريقة 'torch.argmax ()':
يستورد شعلةالخطوة 2: إنشاء موتر ثنائي الأبعاد
ثم استخدم ' الشعلة.الموتر() وظيفة لإنشاء موتر ثنائي الأبعاد وطباعة عناصره. وهنا نقوم بإنشاء ما يلي ' عشرات2 ” 2D tensor:
عشرات2 = شعلة. الموتر ( [ [ 4 , 1 , - 7 ] , [ خمسة عشر , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )مطبعة ( عشرات2 )
أدى هذا إلى إنشاء موتر ثنائي الأبعاد كما هو موضح أدناه:
الخطوة 3: ابحث عن مؤشرات القيمة القصوى
الآن، ابحث عن فهرس القيمة القصوى في ' عشرات2 'الموتر باستخدام' الشعلة.argmax() ' وظيفة:
T2_ind = شعلة. com.argmax ( عشرات2 )الخطوة 4: طباعة فهرس القيمة القصوى
أخيرًا، قم بعرض فهرس القيمة القصوى في موتر الإدخال:
مطبعة ( 'المؤشرات:' , T2_ind )وفقا للمخرجات أدناه، مؤشر القيمة القصوى في ' عشرات2 'الموتر هو '3'. وهذا يعني أن أعلى قيمة للموتر تقع عند المؤشر الثالث وهو ' خمسة عشر ':
الخطوة 5: ابحث عن مؤشرات القيمة القصوى على طول الأعمدة
علاوة على ذلك، يمكن للمستخدمين أيضًا العثور على فهارس/مؤشرات القيم القصوى على طول كل عمود من أعمدة الموتر. على سبيل المثال، يمكننا استخدام ' خافت = 0 'وسيطة مع الدالة 'torch.argmax ()'. يعثر على مؤشرات القيم القصوى على طول الأعمدة في ' عشرات2 'الموتر ثم يطبع تلك المؤشرات:
col_index = شعلة. com.argmax ( عشرات2 , خافت = 0 )مطبعة ( 'المؤشرات في الأعمدة:' , col_index )
يُظهر الناتج أدناه فهارس القيم القصوى على طول كل عمود من أعمدة الموتر:
الخطوة 6: ابحث عن مؤشرات القيمة القصوى على طول الصفوف
وبالمثل، يمكن للمستخدمين أيضًا العثور على فهارس/مؤشرات القيم القصوى على طول كل صف من صفوف الموتر. على سبيل المثال، استخدم ' خافت = 1 'وسيطة مع الدالة 'torch.argmax()' للعثور على مؤشرات القيم القصوى على طول الصفوف في الموتر 'Tens2' ثم طباعة تلك المؤشرات:
Row_index = شعلة. com.argmax ( عشرات2 , خافت = 1 )مطبعة ( 'المؤشرات في الصفوف:' , Row_index )
يمكن رؤية فهارس القيمة القصوى على طول كل صف من موتر 'Tens2' أدناه:
لقد شرحنا بكفاءة طريقة استخدام طريقة 'torch.argmax()' في PyTorch.
ملحوظة : يمكنك الوصول إلى Google Colab Notebook الخاص بنا على هذا وصلة .
خاتمة
لاستخدام طريقة 'torch.argmax()' في PyTorch، قم أولاً باستيراد ' شعلة ' مكتبة. ثم قم بإنشاء الموتر 1D أو 2D المطلوب واعرض عناصره. بعد ذلك، استخدم ' الشعلة.argmax() 'طريقة للعثور على/حساب الفهارس/المؤشرات للقيم القصوى في الموتر. علاوة على ذلك، يمكن للمستخدمين أيضًا العثور على فهارس القيمة القصوى على طول كل صف أو عمود في الموتر باستخدام ' خافت ' دعوى. وأخيرًا، قم بعرض فهرس القيمة القصوى في موتر الإدخال. لقد أوضحت هذه المدونة طريقة استخدام طريقة 'torch.argmax()' في PyTorch.