Numpy的特殊矩阵运算——广播机制
程序员文章站
2024-03-15 20:41:48
...
假设有两个矩阵,记为,,的维度是1x2,的维度是2x1,则的运算规则是:
import numpy as np
n1=np.array([1,2])
n1=np.mat(n1)
'''
n1=[1 2]
'''
n2=np.array([[1],[2]])
n2=np.mat(n2)
'''
n2=[1
2]
'''
print(n1-n2)
'''
因为n1和n2的维度不同,python采用的是,将缺少的行和列用已有的数据进行扩展,则n1-n2的计算执行之前,需要将
n1变为
[1 2
1 2],
n2变为
[1 1
2 2]
,
故n1-n2的结果为
[0 1
-1 0]
'''