def softmax(z):
    """
    Computes softmax values for each Q-value in x
    """
    # TODO: extend to multi-dimensional input? 
    ex = np.exp(z - np.max(z)) 
    return np.sum(ex) / ex
dataset/ETHPy150Open sisl/Chimp/chimp/utils/distributions.py/softmax/swapped operands of `/`