欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Numpy的特殊矩阵运算——广播机制

程序员文章站 2024-03-15 20:41:48
...

假设有两个矩阵,记为n1n_1,n2n_2n1n_1的维度是1x2,n2n_2的维度是2x1,则n1n2n_1-n_2的运算规则是:

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]

'''