Как создать абстрактную элементарную векторную функцию в sympy

#python #sympy #symbolic-math

Вопрос:

Работая с матрицей симпатии, как создать абстрактную векторную функцию по элементам, которая будет использоваться в производных?

Например, если у меня есть:

 u = MatrixSymbol('u', m, 1)
f = u.applyfunc(lambda x: x**2)
 

Тогда f-элементарная функция u с функцией возведения в квадрат по элементам. Как создать его абстрактную версию, такую, чтобы в производной я получал f'(u) вместо (d -> 2d)(u) ?

Я пробовал использовать

 f = Function('f')(u)
 

Но ошибка в производной вместе с матрицей заключалась
в том, что «некоммутативные скаляры в MatMul не поддерживаются». Возможно, это связано с тем, что функции f не было сказано, что она возвращает вектор того же размера.
Есть какие-нибудь решения?