def gval

bend-ml-autograd@0.1.1.0 · main.bend:144

import Base

144def gval(e: G, +env: List<&2, F32>) -> F32:
145  match e:
146    case GCst{c}:
147      c
148    case GVar{i}:
149      lookup(env, i)
150    case GAdd{a, b}:
151      (gval(a, env) + gval(b, env) : F32)
152    case GMul{a, b}:
153      (gval(a, env) * gval(b, env) : F32)
154    case GRelu{a}:
155      F32.max(gval(a, env), 0.0)
156    case GTanh{a}:
157      F32.tanh(gval(a, env))
158    case GExp{a}:
159      F32.exp(gval(a, env))