def drop_mask

bendtorch@1.3.0.0 · main.bend:1254

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

1254def drop_mask(+n: Nat, +c: Nat, +seed: U32, +t: U32, +p: F32) -> TA.Mat<n, c>:
1255  TA.Mat{drop_l(Nat.mul(n, c), 0, mix32(U32.xor(seed, mix32(t))), p, (1.0 / (1.0 - p : F32) : F32), zeros_a(Nat.mul(n, c)))}