def mv

bendtorch@1.3.0.0 · spec.bend:36

import Base

36# M·v: one dot product per row (a linear layer's forward pass, y = W·x)
37def mv(rows: List<&2, List<&2, Nat>>, +v: List<&2, Nat>) -> List<&2, Nat>:
38  match rows:
39    case Nil{}:
40      Nil{}
41    case Con{r, t}:
42      dotn(r, v) <> mv(t, v)