def MatT.rows_take

bend-tensors@0.0.0.2 · bend_tensors.bend:1079

import Base

1079# the first w of every row
1080def MatT.rows_take(+w: Nat, rows: List<&2, List<&2, F32>>) -> List<&2, List<&2, F32>>:
1081  match rows:
1082    case []:
1083      []
1084    case row <> rest:
1085      List.take(&2, F32, row, w) <> MatT.rows_take(w, rest)