def right_walk

bendtorch-core@0.1.0.0 · src/tensor/elementwise.bend:258

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

258def right_walk(-left: List<&2, Nat>, -right: List<&2, Nat>, -output: List<&2, Nat>,
259  map: Shape.Index<left, right, output>, +position: Nat) -> Nat:
260  match map:
261    case Shape.Indexed{rows, exact}:
262      Shape.walk_right(rows, position)
263    case Shape.Unindexed{zero}:
264      0n