def Mat.softmax

bend-ml-tensor@0.1.2.0 · main.bend:400

import Base

400def Mat.softmax(-r: Nat, -c: Nat, a: Mat<r, c>) -> Mat<r, c>:
401  match a:
402    case Mat{x}:
403      Mat{rows.softmax(x)}