Python 如何将这个矩阵相乘？

我如何将下面的矩阵相乘

我想乘以A^2*B，我有一个代码，但它不起作用

将numpy导入为np
A=np.数组（[[0，-1,1]，[3,2,2]，[1,0，-2]]
B=np.数组（[[1,0]，[2,1]，-2,7]]
印刷品（A）
印刷品（B）
C=A*A
印刷品（C）
C*B

+我试着乘以A*B^2，所以我写：

D=A@B@B
印刷品（D）

---

但这与前一个不同。

这是解决方案，如果这是您想要的，请告诉我

import numpy as np

A = np.array([[0,-1,1], [3,2,2], [1,0,-2]])
B = np.array([[1,0], [2,1], [-2,7]])
print (A)
print(B)

C= A @ A @ B
print(C)


[[ 0 -1  1]
 [ 3  2  2]
 [ 1  0 -2]]
[[ 1  0]
 [ 2  1]
 [-2  7]]
[[  2 -30]
 [  4  22]
 [-14  34]]

---

使用，或者，如果您使用的是最新的Python版本（>=3.5），只需使用

@

操作符（

C=a@a@B

）。

*

运算符执行元素乘法（）。是的，谢谢！：）你知道为什么当我想乘的时候A@B@B它不起作用？如果你只是想计算的话A@B@B、 因为尺寸不匹配，所以它不起作用。（看看B@B-这些尺寸已经不合适）。你想要的可能是A@B@B.T