def projected_words

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

import Base

439def projected_words(dims: List<&2, U32>, values: List<&2, U32>) -> List<&2, Nat>:
440  match dims values:
441    case head <> tail value <> rest:
442      broadcast_index(U32.to_nat(head), U32.to_nat(value)) <> projected_words(tail, rest)
443    case dims values:
444      Nil{}