column_num: the number of columns; row_num: the number of rows; k_rep: the number of input matrix; To access more details, see AIE APIs Overview.