def digits

bendtorch-core@0.1.0.0 · src/shape/core.bend:335

import Base

335def digits(+dims: List<&2, Nat>, +index: Nat) -> List<&2, Nat>:
336  match dims:
337    case Nil{}:
338      Nil{}
339    case head <> tail:
340      Nat.div(index, numel(tail)) <> digits(tail, Nat.mod(index, numel(tail)))