def softmax(x): x_exp = np.exp(x) print('x_exp: ', x_exp.shape) ## 输出为(2, 5) 和本来的shape一致 x_sum = np.sum(x_exp, axis=1, keepdims=True) ## keepdims非常重要不然的话x_sum变成rank1 array
numpy softmax广播算法练习
最新推荐文章于 2023-06-14 15:46:24 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)
def softmax(x): x_exp = np.exp(x) print('x_exp: ', x_exp.shape) ## 输出为(2, 5) 和本来的shape一致 x_sum = np.sum(x_exp, axis=1, keepdims=True) ## keepdims非常重要不然的话x_sum变成rank1 array