我正在嘗試按照文檔在此處的建議對(duì) numpy 中的 QR 分解進(jìn)行矢量化，但是我不斷遇到尺寸問題。我對(duì)自己做錯(cuò)了什么感到困惑，因?yàn)槲蚁嘈乓韵聝?nèi)容遵循文檔。有誰知道這有什么問題：import numpy as npX = np.random.randn(100,50,50)vecQR = np.vectorize(np.linalg.qr)vecQR(X)
2 回答
四季花海
TA貢獻(xiàn)1811條經(jīng)驗(yàn) 獲得超5個(gè)贊
來自文檔：“默認(rèn)情況下，假定 pyfunc 將標(biāo)量作為輸入和輸出。”。所以你需要給它一個(gè)簽名：
vecQR = np.vectorize(np.linalg.qr, signature='(m,n)->(m,p),(p,n)')
繁華開滿天機(jī)
TA貢獻(xiàn)1816條經(jīng)驗(yàn) 獲得超4個(gè)贊
np.linalg.qr只映射到 arr 的第一個(gè)軸怎么樣？：
In [35]: np.array(list(map(np.linalg.qr, X)))
Out[35]:
array([[[[-3.30595447e-01, -2.06613421e-02, 2.50135751e-01, ...,
2.45828025e-02, 9.29150994e-02, -5.02663489e-02],
[-1.04193390e-01, -1.95327811e-02, 1.54158438e-02, ...,
2.62127499e-01, -2.21480958e-02, 1.94813279e-01],
[ 1.62712767e-01, -1.28304663e-01, -1.50172509e-01, ...,
1.73740906e-01, 1.31272690e-01, -2.47868876e-01]
添加回答
舉報(bào)
0/150
提交
取消