def ones_then
import bendtorch@1.3.0.0/tensor.bend as Tensorimport 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)