def Vec.softmax.fin

bend-tensors@0.0.0.2 · bend_tensors.bend:1402

import Base

1402def Vec.softmax.fin(+m: Nat, +e: Vec<m>) -> Vec<m>:
1403  Vec.mask(m, Vec.scale(m, F32.div(1.0, Vec.sum(m, e)), e))