בלוג זה ידגים את השיטה לשימוש בשיטת 'torch.argmax()' ב- PyTorch.
כיצד להשתמש בשיטת 'torch.argmax()' ב- PyTorch?
שיטת 'torch.argmax()' לוקחת כל טנסור 1D או 2D כקלט ומחזירה טנזור המכיל את המדדים/אינדקסים של הערכים המקסימליים לאורך הממד הנתון.
התחביר של שיטת 'torch.argmax()' ניתן להלן:
לפיד. argmax ( < input_tensor > )
כדי להשתמש בשיטה זו ב- PyTorch, עברו על הדוגמאות הבאות להבנה טובה יותר:
דוגמה 1: השתמש בשיטת 'torch.argmax()' עם טנסור 1D
בדוגמה הראשונה, ניצור טנזור 1D ונשתמש איתו בשיטת 'torch.argmax()'. בוא נבצע את ההליך שלהלן שלב אחר שלב:
שלב 1: ייבוא ספריית PyTorch
ראשית, ייבא את ' לפיד ' הספרייה להשתמש בשיטת 'torch.argmax()':
יְבוּא לפידשלב 2: צור טנסור 1D
לאחר מכן, צור טנזור 1D והדפיס את האלמנטים שלו. כאן, אנו יוצרים את הדברים הבאים ' עשרות 1 ' טנסור מתוך רשימה באמצעות ' torch.tensor() ' פונקציה:
עשרות 1 = לפיד. מוֹתֵחַ ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )
הדפס ( עשרות 1 )
זה יצר טנזור 1D כפי שניתן לראות להלן:
שלב 3: מצא מדדים של ערך מרבי
כעת, השתמש ב' torch.argmax() פונקציה ' כדי למצוא את האינדקס/המדדים של הערך המקסימלי ב' עשרות 1 'טנסור:
T1_ind = לפיד. argmax ( עשרות 1 )שלב 4: הדפס אינדקס של ערך מרבי
לבסוף, הצג את האינדקס של הערך המרבי בטנזור הקלט:
הדפס ( 'מדדים:' , T1_ind )הפלט שלהלן מציג את האינדקס של הערך המקסימלי ב' עשרות 1 ' טנסור כלומר, 4. זה אומר שהערך הגבוה ביותר של הטנזור נמצא במדד הרביעי שהוא ' 9 ”:
דוגמה 2: השתמש בשיטת 'torch.argmax()' עם טנסור דו-ממדי
בדוגמה השנייה, ניצור טנזור דו-ממדי ונשתמש איתו בשיטת 'torch.argmax()'. בוא נבצע את השלבים שסופקו:
שלב 1: ייבוא ספריית PyTorch
ראשית, ייבא את ' לפיד ' הספרייה להשתמש בשיטת 'torch.argmax()':
יְבוּא לפידשלב 2: צור טנסור דו-ממדי
לאחר מכן, השתמש ב' torch.tensor() פונקציה ליצירת טנזור דו-ממדי ולהדפיס את האלמנטים שלו. כאן, אנו יוצרים את הדברים הבאים ' עשרות2 'טנזור דו מימדי:
עשרות2 = לפיד. מוֹתֵחַ ( [ [ 4 , 1 , - 7 ] , [ חֲמֵשׁ עֶשׂרֵה , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )הדפס ( עשרות2 )
זה יצר טנזור דו-ממדי כפי שניתן לראות להלן:
שלב 3: מצא מדדים של ערך מרבי
כעת, מצא את האינדקס של הערך המקסימלי ב-' עשרות2 טנסור על ידי שימוש ב- torch.argmax() ' פונקציה:
T2_ind = לפיד. argmax ( עשרות2 )שלב 4: הדפס אינדקס של ערך מרבי
לבסוף, הצג את האינדקס של הערך המקסימלי בטנזור הקלט:
הדפס ( 'מדדים:' , T2_ind )על פי הפלט שלהלן, המדד של הערך המרבי ב' עשרות2 'טנסור הוא '3'. זה אומר שהערך הגבוה ביותר של הטנזור הוא במדד השלישי שהוא ' חֲמֵשׁ עֶשׂרֵה ”:
שלב 5: מצא מדדים של ערך מרבי לאורך עמודות
יתר על כן, משתמשים יכולים גם למצוא את האינדקסים/מדדים של הערכים המקסימליים לאורך כל עמודה של טנזור. לדוגמה, אנו יכולים להשתמש ב' עמום=0 ' ארגומנט עם הפונקציה 'torch.argmax()'. הוא מוצא את המדדים של הערכים המקסימליים לאורך העמודות ב' עשרות2 ' טנסור ואז מדפיס את המדדים האלה:
col_index = לפיד. argmax ( עשרות2 , עָמוּם = 0 )הדפס ( 'מדדים בעמודות:' , col_index )
הפלט שלהלן מציג את האינדקסים של הערכים המקסימליים לאורך כל עמודה של הטנזור:
שלב 6: מצא מדדים של ערך מרבי לאורך שורות
באופן דומה, משתמשים יכולים גם למצוא את האינדקסים/מדדים של הערכים המקסימליים לאורך כל שורה של טנזור. לדוגמה, השתמש ב' עמום=1 ' ארגומנט עם הפונקציה 'torch.argmax()' כדי למצוא את המדדים של הערכים המקסימליים לאורך שורות בטנסור 'Tens2' ולאחר מכן להדפיס את המדדים האלה:
row_index = לפיד. argmax ( עשרות2 , עָמוּם = 1 )הדפס ( 'מדדים בשורות:' , row_index )
ניתן לראות את האינדקסים של הערך המרבי לאורך כל שורה של טנזור 'Tens2' להלן:
הסברנו ביעילות את השיטה לשימוש בשיטת 'torch.argmax()' ב- PyTorch.
הערה : תוכל לגשת למחברת Google Colab שלנו כאן קישור .
סיכום
כדי להשתמש בשיטת 'torch.argmax()' ב- PyTorch, ראשית, ייבא את ה-' לפיד 'ספרייה. לאחר מכן, צור את הטנזור 1D או 2D הרצוי והצג את האלמנטים שלו. לאחר מכן, השתמש ב' torch.argmax() ” שיטה למצוא/לחשב את האינדקסים/המדדים של הערכים המקסימליים בטנזור. יתר על כן, משתמשים יכולים גם למצוא את האינדקסים של הערך המרבי לאורך כל שורה או עמודה בטנזור באמצעות ' עָמוּם ' טיעון. לבסוף, הצג את אינדקס הערך המרבי בטנזור הקלט. בלוג זה הדגים את השיטה לשימוש בשיטת 'torch.argmax()' ב- PyTorch.