def ones_then

bendtorch@1.3.0.0 · tensor.bend:52

import bendtorch@1.3.0.0/tensor.bend as Tensor
import Base

52# n copies of 1, then s: s seen with n more leading axes
53def ones_then(n: Nat, s: List<&2, Nat>) -> List<&2, Nat>:
54  match n:
55    case 0n:
56      s
57    case 1n+p:
58      1n <> ones_then(p, s)

Uses (2)

Used by (1)