def heads

bend-ml-tensor@0.1.2.0 · main.bend:127

import Base

127def heads(rows: List<&2, List<&2, F32>>) -> List<&2, F32>:
128  match rows:
129    case Con{Con{+h, t}, rest}:
130      h <> heads(rest)
131    case _:
132      Nil{}