SymPy能简化块矩阵表达式吗？

Question

考虑以下例子：（我试图计算涉及块矩阵的矩阵乘积）

import sympy as sy

k, n = sy.symbols('k,n')
A = sy.MatrixSymbol("A", n, n)
B = sy.MatrixSymbol("B", n, k)
M = sy.BlockMatrix([[A * B, A]])
A.inverse() * M

这将输出

A^-1*Matrix([[A*B, A]])

有没有办法将这个表达式简化/扩展/取消，以更易读的形式Matrix([[B, I]])呈现？

- sarasvati119

Answer 1

是的，使用函数block_collapse可以将块矩阵表达式计算到块级别。

sy.block_collapse(A.inverse()*M)   

returns Matrix([[B, I]])

- user6655984