def index_rows

bendtorch-core@0.1.0.0 · src/shape/core.bend:358

import Base

358def index_rows(+left: List<&2, Nat>, +right: List<&2, Nat>, +output: List<&2, Nat>) -> List<&2, Row>:
359  match left right output:
360    case lh <> lt rh <> rt oh <> ot:
361      Row{numel(ot), broadcast_index(lh, numel(lt)), broadcast_index(rh, numel(rt))} <> index_rows(lt, rt, ot)
362    case left right output:
363      Nil{}