害怕纸飞机

文章
2
资源
0
加入时间
3年0月21天

转换onnx_PyTorch 到 ONNX 到 Tensorflow 到 TFLite的坑

.view()在 Pytorch 中,经常会对某个具体的 Tensor 做变换的处理。但在 PyTorch 转换到 ONNX 的过程中,例如 fea = fea.view(1, -1, 1, 1)ONNX 会错误将 view 的纬度扩张变成 Reshape 节点,如下图所示但是在 Reshape 节点转换到 Tensorflow 的时候,Reshape 节点就变成了 Where 节点和若干节点的组...