[PyTorch] J'étais un peu perdu dans torch.max ()

introduction

Je ne l'ai pas compris dans la documentation PyTorch, alors je vais le laisser. Code de Documentation

Comment utiliser torch.max ()

1. Pour le tenseur 1D

a = torch.randn(1, 3)
a
tensor([[ 0.6763,  0.7445, -2.2369]])
torch.max(a)
tensor(0.7445)

Oui, il renvoie l'élément avec la valeur maximale du tableau unidimensionnel le plus simple

Pour tenseur 2D

a = torch.randn(4, 4)
a
tensor([[-1.2360, -0.2942, -0.1222,  0.8475],
        [ 1.1949, -1.1127, -2.2379, -0.6702],
        [ 1.5717, -0.9207,  0.1297, -1.8768],
        [-0.6172,  1.0036, -0.6060, -0.2432]])
torch.max(a, 1)
torch.return_types.max(values=tensor([0.8475, 1.1949, 1.5717, 1.0036]), indices=tensor([3, 0, 0, 1]))

Je n'ai pas vraiment compris le deuxième argument de ceci C'était l'axe du numpy. Donc personnellement

a = torch.randn(4, 4)
a
tensor([[-1.2360, -0.2942, -0.1222,  0.8475],
        [ 1.1949, -1.1127, -2.2379, -0.6702],
        [ 1.5717, -0.9207,  0.1297, -1.8768],
        [-0.6172,  1.0036, -0.6060, -0.2432]])
axis = 1
torch.max(a, axis)
torch.return_types.max(values=tensor([0.8475, 1.1949, 1.5717, 1.0036]), indices=tensor([3, 0, 0, 1]))

C'est plus facile à comprendre.

torch.max(a, axis)Est-ce utilisé dans la classification des classes, n'est-ce pas?


 Au fait, bien que ce soit pour moi, l'axe est l'axe! (```Axis = 0: col, axis = 1: row```)

# en conclusion
 Je suis toujours influencé par les bibliothèques et les mathématiques, donc je veux pouvoir bien les utiliser le plus tôt possible.


Recommended Posts

[PyTorch] J'étais un peu perdu dans torch.max ()
Dessine un graphique avec Julia ... j'ai essayé une petite analyse
Des mots qui m'intéressaient en tant que débutant en programmation
J'obtiens un UnicodeDecodeError avec mecab-python3
Je me suis perdu dans le labyrinthe
J'obtiens une KeyError dans pyclustering.xmeans
Je veux imprimer dans la notation d'inclusion
J'ai fait un programme de gestion de la paie en Python!
Démarrage de Node.js dans un environnement virtuel
J'ai créé un outil de mot de passe en Python.
Ce à quoi j'étais accro lors de la création d'applications Web dans un environnement Windows
J'ai été en vain parce que je n'ai pas pu obtenir de commande parent avec pybitflyer
Quand j'obtiens une erreur de pilote chrome dans Selenium
Un mémo que j'ai écrit un tri rapide en Python
Python: j'ai pu récurer en lambda
Je veux créer une fenêtre avec Python
J'ai essayé de jouer à un jeu de frappe avec Python
J'ai réfléchi un peu à l'API grandissante de TensorFlow
J'ai écrit une classe en Python3 et Java
J'ai écrit un modèle de conception dans l'édition Kotlin Prototype
J'ai essayé d'ajouter un module Python 3 en C
J'ai écrit un analyseur japonais en japonais en utilisant pyparsing.
J'ai créé un programme cryptographique César en Python.
J'ai fait un peu de recherche sur la classe
J'étais sobrement accro à appeler awscli à partir d'un script Python 2.7 enregistré dans crontab
Je souhaite intégrer une variable dans une chaîne Python
Je veux facilement implémenter le délai d'expiration en python
J'ai écrit un modèle de conception dans l'édition Kotlin Factory
Je veux faire la transition avec un bouton sur le ballon
J'obtiens une exception java.util.regex.PatternSyntaxException lors du fractionnement d'une chaîne dans PySpark
J'avais des ennuis parce que je ne pouvais pas pousser avec heroku
J'ai écrit un modèle de conception dans l'édition Kotlin Builder
[PyTorch] Un peu de compréhension de CrossEntropyLoss avec des formules mathématiques
J'ai écrit un modèle de conception dans l'édition Kotlin Adapter
J'ai essayé d'implémenter un pseudo pachislot en Python
J'ai écrit un modèle de conception en kotlin, édité par Iterator
Je veux travailler avec un robot en python.
Une histoire à laquelle j'étais accro chez np.where
J'ai participé à une programmation compétitive une semaine après avoir commencé la programmation
Impossible de manipuler l'iframe dans la page avec Selenium
Python: peut être répété en lambda
J'ai écrit un modèle de conception dans l'édition de modèle kotlin
J'étais accro à essayer logging.getLogger avec Flask 1.1.x
J'ai implémenté une commande de remplacement de type Vim dans Slackbot #Python
J'ai utilisé bugspots, un outil de prédiction de bogues dans mercurial
Ce à quoi j'étais accro en traitant d'énormes fichiers dans un environnement Linux 32 bits
Une histoire à laquelle j'étais accro en spécifiant nil comme argument de fonction dans Go