softmax.py

def softmax(x):
    """Compute softmax values for each sets of scores in x."""
    e_x = np.exp(x - np.max(x))
    s = e_x / e_x.sum()
    return s