def strided
import bendtorch@1.3.0.0/tensor.bend as Tensorimport 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))