

def argmax(x, axis=(- 1)):
    'Returns the index of the maximum value\n    along a tensor axis.\n    '
    if (axis < 0):
        axis = (axis % len(x.get_shape()))
    return tf.argmax(x, axis)

dataset/ETHPy150Open fchollet/keras/keras/backend/tensorflow_backend.py argmax/original