def column_dims

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul.bend:106

import Base

106def column_dims(+columns: U32, right_vector: Bool) -> List<&2, U32>:
107  match right_vector:
108    case True{}:
109      []
110    case False{}:
111      [columns]