nmi-val / softlearning / utils / numpy.py
numpy.py
Raw
import numpy as np


def softmax(x):
    max_x = np.max(x)
    exp_x = np.exp(x - max_x)
    return exp_x / np.sum(exp_x)