Использование SymPy в ufuncify с комплексными числами
Я заинтересован в создании функции с использованием sympy.utilities.autowrap.ufuncify и использовании ее на комплексных числах. Однако по умолчанию это не работает:
x = sympy.Symbol('x')
f = sympy.utilities.autowrap.ufuncify([x], x**2)
f(arange(5)+1j)
результаты в
TypeError: ufunc 'wrapper_module_0' не поддерживается для типов ввода, и входы не могут быть безопасно приведены к любым поддерживаемым типам в соответствии с правилом приведения "safe"
Это не удивительно, так как f.types является ['d->d'],
Есть ли способ сказать ufuncify для поддержки определенных типов данных (особенно сложных)? У ufuncify есть несколько вариантов, но я не уверен, что кто-то делает то, что я хочу.