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