keras - 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)。这意味着:

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

现在，负数的工作方式与 python 列表、numpy 数组等中的工作方式完全相同。负数表示相反的顺序:

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

当您将axis参数传递给argmax函数时，返回的索引将基于该 Axis 。您的结果将失去这个特定的 Axis ，但保留其他 Axis 。

查看 argmax 将为每个索引返回什么形状:

• K.argmax(a,axis= 0 or -4) 返回 (19,5,80)，其值介于 0 到 18 >
• K.argmax(a,axis= 1 or -3) 返回 (19,5,80)，其值介于 0 到 18 >
• K.argmax(a,axis= 2 or -2) 返回 (19,19,80)，其值介于 0 到 4 >
• K.argmax(a,axis= 3 or -1) 返回 (19,19,5)，其值介于 0 到 79 >