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