当前位置:  开发笔记 > 编程语言 > 正文

如何使用Sympy推导Matrix元素

如何解决《如何使用Sympy推导Matrix元素》经验,为你挑选了1个好方法。

给定矩阵与向量的乘积

平均值

由于A的形状为(m,n),且v为暗n,其中m和n是符号,因此我需要针对矩阵元素计算导数。我还没有找到使用适当向量的方法,所以我从2开始MatrixSymbol

n, m = symbols('n m')
j = tensor.Idx('j')
i = tensor.Idx('i')
l = tensor.Idx('l')
h = tensor.Idx('h')
A = MatrixSymbol('A', n,m)
B = MatrixSymbol('B', m,1)
C=A*B

现在,如果我尝试使用索引从A的元素之一导出,我将得到未评估的表达式:

diff(C, A[i,j])
>>>> Derivative(A*B, A[i, j])

如果我也在C中引入索引(它不会让我仅在结果向量中使用一个索引),我将得到以Sum表示的乘积:

C[l,h]
>>>> Sum(A[l, _k]*B[_k, h], (_k, 0, m - 1))

如果针对矩阵元素派生此结果,则最终得到0而不是带有的表达式KroneckerDelta,这就是我想要得到的结果:

diff(C[l,h], A[i,j])
>>>> 0

我想知道也许我不应该从MatrixSymbols开始。我应该如何去实现我想要的行为?



1> 小智..:

SymPy 尚不知道矩阵演算 ; 特别是不能区分MatrixSymbol物体。您可以使用Matrix填充有符号数组的对象来进行这种计算。缺点是矩阵大小必须明确才能起作用。

例:

from sympy import *
A = Matrix(symarray('A', (4, 5)))
B = Matrix(symarray('B', (5, 3)))
C = A*B
print(C.diff(A[1, 2]))

输出:

Matrix([[0, 0, 0], [B_2_0, B_2_1, B_2_2], [0, 0, 0], [0, 0, 0]])

推荐阅读
手机用户2402851335
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有