def strided

bendtorch@1.3.0.0 · tensor.bend:148

import bendtorch@1.3.0.0/tensor.bend as Tensor
import Base

148def strided(axs: List<&2, Ax>, +j: U32, acc: U32) -> U32:
149  match axs:
150    case Nil{}:
151      acc
152    case Con{Ax{+n, +st}, r}:
153      strided(r, U32.div(j, n), (acc + (U32.mod(j, n) * st : U32) : U32))

Uses (7)

Used by (1)