TensorFlow函数教程：tf.keras.backend.argmax

tf.keras.backend.argmax函数

tf.keras.backend.argmax(
    x,
    axis=-1
)

定义在：tensorflow/python/keras/backend.py。

返回沿轴的最大值索引。

参数：

• x：张量或变量。
• axis：沿其执行缩小的轴。

返回：

该函数返回一个张量。