【onnx】——ScatterND算子

问题描述

onnx模型中有ScatterND算子

---

原因分析：

在forward中，有以下类似的代码时

predictions[:, :, 2:5] += reg[:, :, :3]

出现ScatterND算子

注意：单独的切片操作并不会产生ScatterND

解决方案：

就上述代码来说，将reg[:, :, :3]之前进行切片拆分，如：

reg_0 = reg[:, :, 0]
...

最终存放的predictions[:, :, 2:5]也提前拆分，如：

predictions_0 = predictions[:, :, 0]
...

最终改写为：

predictions_0 += reg_0
...

注意将代码中其余对应的部分修改！