def right_walk

bendtorch-core@0.1.0.0 · src/tensor/contracts/broadcast.bend:17

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

17def right_walk(-left: List<&2, Nat>, -right: List<&2, Nat>, -output: List<&2, Nat>,
18  map: Shape.Index<left, right, output>, +index: Nat) -> Nat:
19  match map:
20    case Shape.Indexed{rows, exact}:
21      Shape.walk_right(rows, index)
22    case Shape.Unindexed{zero}:
23      0n