onnx/tensorflow-onnx

Registering operator for tf.linalg.eig

frytoli opened this issue · 0 comments

Eig

tf.linalg.eig

Describe the operator

Computes the eigen decomposition of a batch of matrices. Very similar to numpy.linalg.eig.

Do you know if this operator be constructed using existing ONNX operators?

I do not believe it can be.

Is this operator used by any model currently? Which one?

N/A

Are you willing to contribute it? (Y/N)

Y

Notes

It looks like MatrixInverse (tf.linalg.inv) has been implemented in Microsoft's extra opset (as discussed in #1617 ). I don't think Eig would be much different of an implementation, but I haven't been able to figure out how MatrixInverse has been mapped (I've seen tf2onnx_custom_ops_tutorial.ipynb and still don't quite get it.). Could you provide any more specific guidance for this op?