Problem 2669. Assign matrix rows/columns to separate variables

This is a slight variant of this problem. Given an input array, return each row/column of the array as a separate variable. The second input variable determines whether to split on columns (dim=1) or rows (dim=2). If the number of output arguments (k) is less than the corresponding array size, only return the first k rows/columns.

Example

If

A = [1  2  3  4
     5  6  7  8
     9 10 11 12];

then

[B,C,D] = matsplit(A,1)
[E,F] = matsplit(A,2)

returns

B = [1 5 9]';
C = [2 6 10]';
D = [3 7 11]';
E = [1 2 3 4];
F = [5 6 7 8];