def index_rows_words
bendtorch-core@0.1.0.0 · src/shape/core.bend:475
import bendtorch-core@0.1.0.0/src/shape/core.bend as Core
import Base
475def index_rows_words(left: List<&2, U32>, right: List<&2, U32>, output: List<&2, U32>) -> List<&2, Row>:
476 match left right output:
477 case lh <> +lt rh <> +rt oh <> +ot:
478 Row{numel_words(ot), broadcast_index(U32.to_nat(lh), numel_words(lt)), broadcast_index(U32.to_nat(rh), numel_words(rt))} <> index_rows_words(lt, rt, ot)
479 case left right output:
480 Nil{}