def strided

bendtorch-core@0.1.0.0 · src/autograd/reduce.bend:131

import bendtorch-core@0.1.0.0/src/autograd/reduce.bend as Reduce

import Base

131def strided(fuel: Nat, +step: Nat, +base: Nat, +digit: Nat, +lift: Nat, accumulator: F32, array: Array<F32>) -> F32 & Array<F32>:
132  match fuel:
133    case 0n:
134      (accumulator, array)
135    case 1n+remaining:
136      strided_run(remaining, step, base, digit, lift, accumulator, Array.get(F32, array, U32.from_nat(strided_offset(step, base, digit, lift))))

Uses (9)

Used by (4)