Numpy 广播机制 不同shape的数组进行计算

所属的课程名称及链接

环境信息

•  ModelArts
  •    Notebook - Multi-Engine 2.0 (python3)
    •      JupyterLab - Notebook - Conda-python3
      •       numpy 1.19.1

Numpy 广播机制 不同shape的数组进行计算

a = np.array([1,2,3])
b = a.reshape(3,1)
print("a",a)
print("b",b)
c = a*b  # 广播
print("c",c)
print("----")
print(c[0,0], a[0], b[0][0])
print(c[0,1], a[1], b[0][0])

doc="""
# 广播后
a_ex
1 2 3
1 2 3
1 2 3

b_ex
1 1 1
2 2 2
3 3 3

c[0][0] = a_ex[0][0] * b_ex[0][0]  # 对应位相乘
"""

a [1 2 3]
b [[1]
 [2]
 [3]]
c [[1 2 3]
 [2 4 6]
 [3 6 9]]
----
1 1 1
2 2 1

备注