def softmax(x):
e_x = T.exp(x - x.max(axis=0, keepdims=True))
out = e_x / e_x.sum(axis=0, keepdims=True)
return out
微信
公众号
扫码关注公众号
微信
公众号