how to deal with numpy stack code example

Example: numpy stack in new dimension

np.stack((R_mat, G_mat, B_mat), axis = 2)
# identify any new dimension as you like