【onnx】——ScatterND算子:从PyTorch切片赋值到ONNX模型部署的桥梁
1. ScatterND算子的核心作用:PyTorch到ONNX的桥梁
在PyTorch模型转ONNX格式时,经常会遇到一个看似简单却暗藏玄机的操作——张量切片赋值。比如下面这段代码:
x torch.randn(20, 200, 200)
y torch.randn(10, 200, 200)
x[0:10, :, :] y…
2026/6/30 14:56:08