[ONNX][Complex] Support view_as_complex

[iou2much]

🚀 Feature

would you support view_as_complex for converting to onnx model? Thanks a lot

Motivation

when I try to convert a nn contains complex tensor computation, it says:

Pytorch export onnx: RuntimeError Exporting the operator view_as_complex to ONNX opset version 9 is not supported. Please open a bug to request ONNX export support for the missing operator

also : onnx/onnx#3173

cc @houseroad @spandantiwari @lara-hdr @BowenBao @neginraoof

[iou2much]

I'm also trying to do it myself, following this doc, but I'm quite new to pytorch. I'm not sure how to implement a new method which is not standardized in Onnx.

[spandantiwari]

@iou2much - We can put this on the backlog, but we do not have immediate plans for adding support for this. Please consider sending a PR for adding support. If you are trying to export this as custom op, I think this doc could be useful.
https://github.com/onnx/tutorials/tree/master/PyTorchCustomOperator

[iou2much]

Thank you. Let me try

[urinieto]

Is there still not enough interest to support this? Note that the STFT ONNX operator on PyTorch requires this and view_as_real for full compatibility (#92087).

[justinchuby]

@urinieto we will need new onnx operators to support this. I can create an issue when I’m back

[justinchuby]

Complex number support will be handled via the torch.onnx.dynamo_export exporter.

[mush42]

When will this be available in dev builds?

[justinchuby]

It is available. You can use the latest onnxscript

[jinhonglu]

It seems that there is still problem of exporting complex number with dynamo_export

#121100