def argmax()

in maddpg/common/tf_util.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

def argmax(x, axis=None):
    return tf.argmax(x, axis=axis)