# logits 的最后一维=类别数
def sofmax(logits):
e_x = np.exp(logits)
probs = e_x / np.sum(e_x, axis=-1, keepdims=True)
return probs
numpy softmax
最新推荐文章于 2024-02-02 09:53:12 发布
# logits 的最后一维=类别数
def sofmax(logits):
e_x = np.exp(logits)
probs = e_x / np.sum(e_x, axis=-1, keepdims=True)
return probs