1. 程式人生 > >numpy隨筆-廣播

numpy隨筆-廣播

import numpy as np
x=np.array([[1,2,3],[4,5,6]])
y=np.array([10,20,30])
print(x)
print("---")
print(y)
print("---")
x+y

較小的張量會被廣播,以匹配較大張量的形狀。沿軸擴充套件。

[[1 2 3]
 [4 5 6]]
---
[10 20 30]
---
Out[13]:
array([[11, 22, 33],
       [14, 25, 36]])