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