def Mat.softmax_rows

bend-tensors@0.0.0.2 · bend_tensors.bend:1540

import Base

1540# softmax along each row over its cols columns
1541def Mat.softmax_rows(+rows: Nat, +cols: Nat, +m: Mat<rows, cols>) -> Mat<rows, cols>:
1542  Mat.softmax_rows.fin(rows, cols, Mat.exp(rows, cols, Mat.sub_col(rows, cols, m, Mat.max_rows(rows, cols, m))))