def RestWalk

bendtorch-core@0.1.0.0 · src/autograd/reduce_fiber_proof.bend:642

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

642# The walk below a row, at base 0, visits the masked positions below it.
643def RestWalk(+rest: List<&2, Shape.Row>, +others: List<&2, Nat>, +inner: Nat, +stride: Nat) -> Data:
644  {masked(stride, rest, inner, 0n, 0n) ==
645    visited(rest, others, Reduce.first_span(rest, others), inner, 0n, Reduce.first_digit(rest, inner)) : List<&2, Nat>}