import numpy as np # 一维数组 arr = np.array([1, 2, 3]) # 标量5自动"广播"成[5,5,5]，再相加 result = arr + 5 print(result) # 输出 [6 7 8]
import numpy as np # 二维数组（2行3列） arr2d = np.array([[1, 2, 3], [4, 5, 6]]) # 一维数组（1行3列），自动广播成2行3列（复制一行） arr1d = np.array([10, 20, 30]) result = arr2d + arr1d print(result) # 输出： # [[11 22 33] # [14 25 36]]
import numpy as np # 列向量（2行1列） col = np.array([[10], [20]]) # 行向量（1行3列） row = np.array([1, 2, 3]) # 列向量广播成2行3列，行向量也广播成2行3列，再相加 result = col + row print(result) # 输出： # [[11 12 13] # [21 22 23]]
广播的作用就是让张量补全进行计算 ，增加兼容性