def softmax()

in src/model.py [0:0]

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

def softmax(x, axis=-1):
    x = x - tf.reduce_max(x, axis=axis, keepdims=True)
    ex = tf.exp(x)
    return ex / tf.reduce_sum(ex, axis=axis, keepdims=True)