def Mat.relu

bend-ml-tensor-array@0.1.19.0 · main.bend:1343

import Base

1343def Mat.relu(+r: Nat, +c: Nat, a: Mat<r, c>) -> Mat<r, c>:
1344  match a:
1345    case Mat{d}:
1346      relu_fin(r, c, relu_l(Nat.mul(r, c), 0, Array.get(F32, d, 0)))