def split_dims
bendtorch-core@0.1.0.0 · src/tensor/matmul.bend:744
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matmul
import Base
744def split_dims(+order: Order, dims: List<&2, U32>) -> Result<&2, &2, Error, Split<order, dims>>:
745 match order:
746 case Kept{}:
747 match dims:
748 case +a <> +b <> Nil{}:
749 Done{Split{a, b, {==}}}
750 case rest:
751 Fail{RankMismatch{}}
752 case Swapped{}:
753 match dims:
754 case +a <> +b <> Nil{}:
755 Done{Split{b, a, {==}}}
756 case rest:
757 Fail{RankMismatch{}}