def MatT.chunks

bend-tensors@0.0.0.2 · bend_tensors.bend:1155

import Base

1155# xs cut into n rows of w
1156def MatT.chunks(+n: Nat, +w: Nat, +xs: List<&2, F32>) -> List<&2, List<&2, F32>>:
1157  match n:
1158    case 0n:
1159      []
1160    case 1n+p:
1161      List.take(&2, F32, xs, w) <> MatT.chunks(p, w, List.drop(&2, F32, xs, w))