def Mat.softmax_rows.fin

bend-tensors@0.0.0.2 · bend_tensors.bend:1537

import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
import Base

1537def Mat.softmax_rows.fin(+rows: Nat, +cols: Nat, +e: Mat<rows, cols>) -> Mat<rows, cols>:
1538  Mat.mask(rows, cols, Mat.scale_rows(rows, cols, e, Vec.map(~(s => F32.div(1.0, s)), rows, Mat.sum_rows(rows, cols, e))))

Uses (7)

Used by (1)