keras.argmax中axis = -1的含义是什么？

Question

我是Keras的初学者，需要帮助才能理解keras.argmax（a，axis = -1）和keras.max（a，axis = -1）。当a.shape =（19,19,5,80）？

时，axis = -1的含义是什么？

如果你能回答keras.argmax（a，axis = -1）和keras.max（a，axis = -1）

的输出，我将不胜感激。

提前致谢

- 年轻

Answer 1

这意味着argmax返回的索引将取自最后一个轴。

您的数据有一些形状(19,19,5,80)。这意味着：

• 轴0 = 19个元素
• 轴1 = 19个元素
• 轴2 = 5个元素
• 轴3 = 80个元素

现在，负数与python列表，numpy数组等完全相同。负数表示逆序：

• 轴-1 = 80个元素
• 轴-2 = 5个元素
• 轴-3 = 19个元素
• 轴-4 = 19个元素

将axis参数传递给argmax函数时，返回的索引将基于此轴。您的结果将丢失此特定轴，但保留其他轴。

查看每个索引将返回argmax的形状：

• K.argmax(a,axis= 0 or -4)使用(19,5,80)
中的值返回0 to 18
• K.argmax(a,axis= 1 or -3)使用(19,5,80)
中的值返回0 to 18
• K.argmax(a,axis= 2 or -2)使用(19,19,80)
中的值返回0 to 4
• K.argmax(a,axis= 3 or -1)使用(19,19,5)
中的值返回0 to 79