Registering operator for tf.linalg.eig
frytoli opened this issue · 0 comments
frytoli commented
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?