def outer

bendtorch@1.3.0.0 · spec.bend:52

import Base

52# g ⊗ v: row i is g[i]·v (a linear layer's weight gradient, dW = dy ⊗ x)
53def outer(g: List<&2, Nat>, +v: List<&2, Nat>) -> List<&2, List<&2, Nat>>:
54  match g:
55    case Nil{}:
56      Nil{}
57    case Con{gi, gt}:
58      vscale(gi, v) <> outer(gt, v)