我們可以對這個 for 回圈進行矢量化,最終在每次迭代中產生更新并在 for 中使用結果,如下所示:
x = np.array([[1,2],[2,3],[3,4]])
y = np.array([1,-1,1])
z = np.array([-1,1,-1])
w = 2
out = np.array([10,20])
for xi,yi,zi in zip(x, y, z):
out = out (w*(yi-zi))*xi
輸出:
>>> out
array([ 18, 32])
uj5u.com熱心網友回復:
你可以簡單地使用:
(w*(y-z)*x.T).sum(1)
輸出: array([ 8, 12])
添加現有陣列:
out = np.array([10,20])
out = (w*(y-z)*x.T).sum(1)
輸出: array([18, 32])
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/316902.html
