def softmax(X):
Xexp = X.exp()
partition = Xexp.sum(axis=1, keepdims=True)
return Xexp / partition # 这里应用了广播机制
softmax()定义
最新推荐文章于 2025-04-26 19:00:00 发布
def softmax(X):
Xexp = X.exp()
partition = Xexp.sum(axis=1, keepdims=True)
return Xexp / partition # 这里应用了广播机制