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.