Πώς να χρησιμοποιήσετε τη μέθοδο 'torch.argmax()' στο PyTorch;

Στο PyTorch, το ' torch.argmax() Η μέθοδος είναι μια ενσωματωμένη συνάρτηση που επιστρέφει δείκτες μέγιστων τιμών ενός συγκεκριμένου τανυστή σε μια δεδομένη διάσταση. Οι χρήστες χρησιμοποιούν αυτή τη συνάρτηση όταν εργάζονται με τανυστές και θέλουν να βρουν τον δείκτη της μέγιστης τιμής κατά μήκος της δεδομένης διάστασης ενός τανυστή. Επιπλέον, αυτή η μέθοδος μπορεί επίσης να είναι χρήσιμη για ταξινόμηση όπου οι χρήστες θέλουν να γνωρίζουν ποια κλάση έχει τη μεγαλύτερη πιθανότητα.

Αυτό το ιστολόγιο θα αποτελεί παράδειγμα της μεθόδου χρήσης της μεθόδου 'torch.argmax()' στο PyTorch.

Πώς να χρησιμοποιήσετε τη μέθοδο 'torch.argmax()' στο PyTorch;

Η μέθοδος 'torch.argmax()' παίρνει οποιονδήποτε 1D ή 2D τανυστή ως είσοδο και επιστρέφει έναν τανυστή που περιέχει τους δείκτες/δείκτες των μέγιστων τιμών κατά μήκος της δεδομένης διάστασης.

Η σύνταξη της μεθόδου 'torch.argmax()' δίνεται παρακάτω:

δάδα. argmax ( < input_tensor > )

Για να χρησιμοποιήσετε αυτήν τη μέθοδο στο PyTorch, διαβάστε τα ακόλουθα παραδείγματα για καλύτερη κατανόηση:

Παράδειγμα 1: Χρησιμοποιήστε τη μέθοδο 'torch.argmax()' με 1D Tensor

Στο πρώτο παράδειγμα, θα δημιουργήσουμε έναν τανυστή 1D και θα χρησιμοποιήσουμε τη μέθοδο 'torch.argmax()' μαζί του. Ας ακολουθήσουμε την παρακάτω διαδικασία βήμα προς βήμα:

Βήμα 1: Εισαγάγετε τη βιβλιοθήκη PyTorch

Πρώτα, εισαγάγετε το ' δάδα ” βιβλιοθήκη για να χρησιμοποιήσετε τη μέθοδο “torch.argmax()”:

εισαγωγή δάδα

Βήμα 2: Δημιουργήστε 1D Tensor

Στη συνέχεια, δημιουργήστε έναν τανυστήρα 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. Σημαίνει ότι η υψηλότερη τιμή του τανυστή είναι στον 4ο δείκτη που είναι ' 9 ”:

Παράδειγμα 2: Χρησιμοποιήστε τη μέθοδο 'torch.argmax()' με 2D Tensor

Στο δεύτερο παράδειγμα, θα δημιουργήσουμε έναν τανυστή 2D και θα χρησιμοποιήσουμε τη μέθοδο 'torch.argmax()' μαζί του. Ας ακολουθήσουμε τα προβλεπόμενα βήματα:

Βήμα 1: Εισαγάγετε τη βιβλιοθήκη PyTorch

Πρώτα, εισαγάγετε το ' δάδα ” βιβλιοθήκη για να χρησιμοποιήσετε τη μέθοδο “torch.argmax()”:

εισαγωγή δάδα

Βήμα 2: Δημιουργήστε 2D Tensor

Στη συνέχεια, χρησιμοποιήστε το ' torch.tensor() Λειτουργία για τη δημιουργία ενός δισδιάστατου τανυστή και την εκτύπωση των στοιχείων του. Εδώ, δημιουργούμε το εξής ' Tens2 «2D τανυστής:

Tens2 = δάδα. τανύων μύς ( [ [ 4 , 1 , - 7 ] , [ δεκαπέντε , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )

Τυπώνω ( Tens2 )

Αυτό έχει δημιουργήσει έναν δισδιάστατο τανυστή όπως φαίνεται παρακάτω:

Βήμα 3: Βρείτε δείκτες μέγιστης αξίας

Τώρα, βρείτε τον δείκτη της μέγιστης τιμής στο ' Tens2 ' τανυστής χρησιμοποιώντας το ' torch.argmax() ' λειτουργία:

T2_ind = δάδα. argmax ( Tens2 )

Βήμα 4: Εκτύπωση ευρετηρίου μέγιστης τιμής

Τέλος, εμφανίστε τον δείκτη της μέγιστης τιμής στον τανυστή εισόδου:

Τυπώνω ( 'Δείκτες:' , T2_ind )

Σύμφωνα με την παρακάτω έξοδο, ο δείκτης της μέγιστης τιμής στο « Tens2 'Ο τανυστής είναι '3'. Σημαίνει ότι η υψηλότερη τιμή του τανυστή είναι στον 3ο δείκτη που είναι ' δεκαπέντε ”:

Βήμα 5: Βρείτε δείκτες μέγιστης τιμής κατά μήκος των στηλών

Επιπλέον, οι χρήστες μπορούν επίσης να βρουν τους δείκτες/δείκτες των μέγιστων τιμών κατά μήκος κάθε στήλης ενός τανυστή. Για παράδειγμα, μπορούμε να χρησιμοποιήσουμε το « dim=0 όρισμα με τη συνάρτηση 'torch.argmax()'. Βρίσκει τους δείκτες των μέγιστων τιμών κατά μήκος των στηλών στο ' Tens2 ' τανυστής και στη συνέχεια εκτυπώνει αυτούς τους δείκτες:

col_index = δάδα. argmax ( Tens2 , αμυδρός = 0 )

Τυπώνω ( 'Δείκτες σε στήλες:' , col_index )

Η παρακάτω έξοδος δείχνει τους δείκτες των μέγιστων τιμών κατά μήκος κάθε στήλης του τανυστή:

Βήμα 6: Βρείτε δείκτες μέγιστης τιμής κατά μήκος των σειρών

Ομοίως, οι χρήστες μπορούν επίσης να βρουν τους δείκτες/δείκτες των μέγιστων τιμών κατά μήκος κάθε σειράς ενός τανυστή. Για παράδειγμα, χρησιμοποιήστε το ' dim=1 όρισμα με τη συνάρτηση 'torch.argmax()' για να βρείτε τους δείκτες των μέγιστων τιμών κατά μήκος των σειρών στον τανυστήρα 'Tens2' και στη συνέχεια να εκτυπώσετε αυτούς τους δείκτες:

row_index = δάδα. argmax ( Tens2 , αμυδρός = 1 )

Τυπώνω ( 'Δείκτες σε σειρές:' , row_index )

Οι δείκτες των μέγιστων τιμών κατά μήκος κάθε σειράς ενός τανυστή 'Tens2' φαίνονται παρακάτω:

Έχουμε εξηγήσει αποτελεσματικά τη μέθοδο χρήσης της μεθόδου 'torch.argmax()' στο PyTorch.

Σημείωση : Μπορείτε να αποκτήσετε πρόσβαση στο Σημειωματάριο Google Colab από αυτό Σύνδεσμος .

συμπέρασμα

Για να χρησιμοποιήσετε τη μέθοδο 'torch.argmax()' στο PyTorch, πρώτα εισαγάγετε το ' δάδα ' βιβλιοθήκη. Στη συνέχεια, δημιουργήστε τον επιθυμητό τανυστή 1D ή 2D και δείτε τα στοιχεία του. Στη συνέχεια, χρησιμοποιήστε το ' torch.argmax() ” μέθοδος εύρεσης/υπολογισμού των δεικτών/δεικτών των μέγιστων τιμών στον τανυστή. Επιπλέον, οι χρήστες μπορούν επίσης να βρουν τους δείκτες των μέγιστων τιμών κατά μήκος κάθε γραμμής ή στήλης στον τανυστή χρησιμοποιώντας το ' αμυδρός ' διαφωνία. Τέλος, εμφανίστε τον δείκτη της μέγιστης τιμής στον τανυστή εισόδου. Αυτό το ιστολόγιο έχει ως παράδειγμα τη μέθοδο χρήσης της μεθόδου 'torch.argmax()' στο PyTorch.

Πώς να χρησιμοποιήσετε τη μέθοδο 'torch.argmax()' στο PyTorch;