def drop_fwd

bendtorch@1.3.0.0 · main.bend:1262

import bendtorch@1.3.0.0/main.bend as Bendtorch

import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA

1262def drop_fwd(+n: Nat, +c: Nat, +seed: U32, +p: F32, d: DropP, x: TA.Mat<n, c>) -> Fwd<DropP, TA.Mat<n, c>, TA.Mat<n, c>>:
1263  match d:
1264    case DropP{+t}:
1265      drop_f2(n, c, t, Ew.zip(~f32_mul, n, c, x, drop_mask(n, c, seed, t, p)))

Uses (10)

Used by (1)

bendtorch@1.3.0.0/main:Dropout