tf.keras.backend.argmax(
x,
axis=-1
)
Defined in tensorflow/python/keras/backend.py.
Returns the index of the maximum value along an axis.
Arguments:
x: Tensor or variable.axis: axis along which to perform the reduction.
Returns:
A tensor.