def offset

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

import bendtorch-core@0.1.0.0/src/shape/core.bend as Core
import Base

328def offset(+dims: List<&2, Nat>, values: List<&2, Nat>) -> Nat:
329  match dims values:
330    case head <> tail value <> rest:
331      Nat.add(Nat.mul(value, numel(tail)), offset(tail, rest))
332    case dims values:
333      0n

Uses (5)

Used by (22)