97 votes

Quelle est la signification de axis = -1 dans keras.argmax?

Je suis débutant à Keras et j'ai besoin d'aide pour comprendre keras.argmax(a, axis=-1) et keras.max(a, axis=-1) . Quelle est la signification de axis=-1 lorsque a.shape = (19, 19, 5, 80) ? Et aussi quelle sera la sortie de keras.argmax(a, axis=-1) et keras.max(a, axis=-1) ?

187voto

Daniel Points 2149

Cela signifie que l'indice qui sera retournée par argmax seront prises à partir de la dernière axe.

Vos données ont une certaine forme (19,19,5,80). Cela signifie:

  • L'axe 0 = 19 éléments
  • Axe 1 = 19 éléments
  • Axe 2 = 5 éléments
  • Axe 3 = 80 éléments

Maintenant, les nombres négatifs fonctionnent exactement comme en python listes, des tableaux numpy, etc. Les nombres négatifs représentent l'ordre inverse:

  • Axe -1 = 80 éléments
  • Axe -2 = 5 éléments
  • Axe -3 = 19 éléments
  • Axe -4 = 19 éléments

Lorsque vous passez à l' axis paramètre à l' argmax de la fonction, les indices retourné sera basé sur cet axe. Vos résultats vont perdre cette spécifique des axes, mais gardez les autres.

Voir quelle forme argmax sera de retour pour chaque indice:

  • K.argmax(a,axis= 0 or -4) retours (19,5,80) avec des valeurs de 0 to 18
  • K.argmax(a,axis= 1 or -3) retours (19,5,80) avec des valeurs de 0 to 18
  • K.argmax(a,axis= 2 or -2) retours (19,19,80) avec des valeurs de 0 to 4
  • K.argmax(a,axis= 3 or -1) retours (19,19,5) avec des valeurs de 0 to 79

Prograide.com

Prograide est une communauté de développeurs qui cherche à élargir la connaissance de la programmation au-delà de l'anglais.
Pour cela nous avons les plus grands doutes résolus en français et vous pouvez aussi poser vos propres questions ou résoudre celles des autres.

Powered by:

X