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]
广播的作用就是让张量补全进行计算 ，增加兼容性