Broadcasting and the softmax function in Python

最新推荐文章于 2024-07-10 23:11:48 发布

weixin_45405128

最新推荐文章于 2024-07-10 23:11:48 发布

阅读量309

点赞数 1

分类专栏： Python

本文链接：https://blog.csdn.net/weixin_45405128/article/details/102329179

版权

Python 专栏收录该内容

81 篇文章 1 订阅

订阅专栏

def softmax(x):
    x_exp = np.exp(x)
    x_sum = np.sum(x_exp,axis=1,keepdims=True)
    s = x_exp/x_sum    
    return s

x = np.array([
    [9, 2, 5, 0, 0],
    [7, 5, 0, 0 ,0]])
print("softmax(x) = " + str(softmax(x)))

OUT:

softmax(x) = [[  9.80897665e-01   8.94462891e-04   1.79657674e-02   1.21052389e-04     1.21052389e-04]
              [  8.78679856e-01   1.18916387e-01   8.01252314e-04   8.01252314e-04     8.01252314e-04]]