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

2019-03-06 16:10 更新

tf.keras.backend.argmax函数

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

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

返回沿轴的最大值索引。

参数：

该函数返回一个张量。