基于numy实现softmax功能

本文利用numpy模块实现softmax的功能。

    import numpy as np

    def softmax_np(logits):
        assert (isinstance(logits, np.ndarray)), 'only numpy is available'
        exp_value = np.exp(logits)  # 计算指数值
        dim_ext = np.sum(exp_value, 1).reshape(-1, 1)
        return exp_value / dim_ext

    x_val = [[1, 2, 3], [3, 2, 2]]
    logits = np.array(x_val)
    s_v = softmax_np(logits)
    print s_v, np.sum(s_v, 1)

参考文献：

1. https://stackoverflow.com/questions/34968722/how-to-implement-the-softmax-function-in-python

2. softmax函数计算时候为什么要减去一个最大值？