比如一个(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:]))
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
