比如一个(4,100,224,224,3)的数组想将维度前两个维度合并,变成(400,224,224,3),可以直接
new_array = ori_array.reshape([400,224,224,3])
为了推广到数组各个维度未知的情况,可以改成
new_array = ori_array.reshape([ori_array.shape[0] * ori_array.shape[1]] + list(ori_array.shape[2:]))
比如一个(4,100,224,224,3)的数组想将维度前两个维度合并,变成(400,224,224,3),可以直接
new_array = ori_array.reshape([400,224,224,3])
为了推广到数组各个维度未知的情况,可以改成
new_array = ori_array.reshape([ori_array.shape[0] * ori_array.shape[1]] + list(ori_array.shape[2:]))