numpy 实现 softmax

本文介绍了一种列向量的softmax函数实现方法,通过防止数值上溢来确保计算的稳定性。并提供了一个示例,展示了如何使用该函数对一个包含20个元素的列向量进行softmax计算。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

def col_softmax(x):
    x = np.asarray(x)
    x_col_max = x.max(axis=0)
    x_col_max = x_col_max.reshape([1,x.shape[1]])
    x = x - x_col_max  # 防止数值上溢
    x_exp = np.exp(x)
    x_exp_col_sum = x_exp.sum(axis=0).reshape([1,x.shape[1]])
    softmax = x_exp / x_exp_col_sum
    return softmax

测试

a = [[i] for i in range(20)]
col_softmax(a)
'''
array([[3.54164282e-09],
       [9.62718333e-09],
       [2.61693975e-08],
       [7.11357977e-08],
       [1.93367146e-07],
       [5.25626400e-07],
       [1.42880069e-06],
       [3.88388295e-06],
       [1.05574885e-05],
       [2.86982290e-05],
       [7.80098745e-05],
       [2.12052824e-04],
       [5.76419339e-04],
       [1.56687021e-03],
       [4.25919483e-03],
       [1.15776919e-02],
       [3.14714295e-02],
       [8.55482150e-02],
       [2.32544158e-01],
       [6.32120560e-01]])
'''
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

颹蕭蕭

白嫖?

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值