Merge pull request #18048 from l-bat:onnx_torchvision

2026-07-29 23:33:05 +04:00 · 2020-08-06 20:21:47 +00:00
parent 1067cd0649 6226ea0085
commit 336627a776
1 changed files with 1 additions and 0 deletions
@@ -1395,6 +1395,7 @@ void ONNXImporter::populateNet(Net dstNet)

                    inpShape.erase(inpShape.begin() + axis);
                    layerParams.type = "Reshape";
+                    layerParams.set("axis", 0);
                    layerParams.set("dim", DictValue::arrayInt(&inpShape[0], inpShape.size()));
                    node_proto.set_input(0, sliceLp.name);
                }