广播机制(broadcast)是一种重要机制,有了这种机制,就可以方便地对不同维的张量进行运算

    比如:

    1. c=np.arange(10).reshape([5,2])
    2. d=np.arange(2).reshape([1,2])

    c:

    1. array([[0, 1],
    2. [2, 3],
    3. [4, 5],
    4. [6, 7],
    5. [8, 9]])

    d:

    1. array([[0, 1]])
    1. c+d # 首先将d数组进行复制扩充为[5,2],然后相加。

    运算结果:

    1. array([[ 0, 2],
    2. [ 2, 4],
    3. [ 4, 6],
    4. [ 6, 8],
    5. [ 8, 10]])

    原理:
    📃 NumPy广播 - 图1