CoreML 如何将(1,3,80,80,19)数组重塑为(3,6400,19)或(1,3,6400,19)？

• 核心配置错误：add_reshape接口默认mode=0，该模式强制要求输出张量的秩必须和输入完全一致，不允许增删维度。你的输入是5维张量，目标输出是3/4维，mode=0会触发CoreML自动拆分、补全维度以凑够5维的要求，最终生成完全不符合预期的reshape目标，触发计算核参数校验报错。
• 轴序对应规则：CoreML底层计算核对张量形状的存储为逆序排列，你定义的输入形状(1,3,80,80,19)传递到底层时会被转成(19,80,80,3,1)，和报错信息中的输入形状完全匹配。

按照你需要的两种输出形状，对应修改代码即可，核心是把reshape的mode改为1（允许调整输出张量的秩）：

输出为(3, 6400, 19)（去掉开头的批次1维） #

直接传入对应目标形状，也可以用-1代替需要合并的维度值，让内核自动推导，避免硬编码计算错误：

builder.add_reshape(
    name='714_reshape',
    input_name='714',
    output_name='714_reshape',
    target_shape=(3, -1, 19),
    mode=1
)

内核会自动将相邻的两个80维度相乘得到6400，且不会打乱原有元素的相对顺序。

输出为(1, 3, 6400, 19)（保留开头的批次1维） #

同样修改mode为1，target_shape写为(1, 3, -1, 19)即可，内核会自动完成两个80维度的合并，总元素数匹配不会触发报错。

小技巧：用-1标记需要自动推导的维度，无需手动计算80*80的乘积，也能规避轴序逆序带来的维度值写错问题。

内容的提问来源于stack exchange，提问作者OMM