def argmax(x, axis=-1):
    '''Returns the index of the maximum value
    along a tensor axis.
    '''
    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