Unfortunately no as this would require to add this specific function to onnx. I am not sure how easy/hard that would be as this might depend on the other framework’s (caffe2 here) capacity to do the exact same operation.