def matrix_dims

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

import Base

113def matrix_dims(+rows: U32, left_vector: Bool, +columns: U32, right_vector: Bool) -> List<&2, U32>:
114  match left_vector:
115    case True{}:
116      column_dims(columns, right_vector)
117    case False{}:
118      rows <> column_dims(columns, right_vector)