I need to do some kind of channel indexation/insertion using scatter functions, but the element-wise scatter available in Pytorch leads to a Scatter ONNX function that is very cumbersome (as it requires indexing every single element of the tensor) while I only need to index channels.
The ONNX function ScatterND would allow that (which would be a lot less memory-expensive, since it requires only storing channels indices) but no equivalent exist in Pytorch (to my knowledge).
I was planning to implement a custom function, that would be converted into a ScatterND, but maybe you know one more straightforward way to get this operator.