def leading

stelliferous@0.0.2.0 · tensor_window.bend:26

import Base
import stelliferous@0.0.2.0/tensor_shape.bend as Shapes

26def leading(ranked: Bool,+shape: List<&2,U32>) -> Maybe<&2,U32>:
27  match ranked:
28    case True{}: Shapes.product(List.take(&2,U32,shape,Nat.sub(List.length(&2,U32,shape),2n)),Some{1})
29    case False{}: None{}