"how to change numpy array from (128,128,3) to (3,128,128)?","a.transpose(2, 1, 0)"