How can I integrate a function scipy.sparse.linalg.eigs inside a tensorflow graph?

Question

I would like to use the function eigs inside a tensorflow graph. I\'ve tried wrapping it using tf.numpy_function. It works fine when I try it on a Tensor object that I created t