STM32CubeIDE部署神经网络.onnx错误INTERNAL ERROR: Unknown shape destinatio
点击对AI算法分析后出错,如下这是PyTorch模型转换到ONNX模型中非常常见的reshape的问题,一个简单的PyTorch中的view操作,期待出来的就是一个简单的reshape操作,但是如果你的形状参数用的是形如x.size(0)这样的话就会出现上图的问题,必须做一个强制类型转换int(x.size(0),或者用onnx-simplifier来处理一下模型,如下:推荐文章:利用pytorc