def pad

bendtorch@1.3.0.0 · tensor.bend:60

import Base

60# s aligned to rank r by leading size-1 axes, as broadcasting aligns shapes from the right
61def pad(+s: List<&2, Nat>, +r: Nat) -> List<&2, Nat>:
62  ones_then(Nat.sub(r, List.length(&2, Nat, s)), s)