假設(shè)我有 3 維 numpy array a，例如如下：import numpy as np
a = np.random.randn(3, 3, 3)如何將（矩陣->標(biāo)量）類型函數(shù)應(yīng)用于a？更具體地說，我想以一種計(jì)算效率更高的方式做如下等效的事情：[np.linalg.det(e) for e in a]
1 回答
慕工程0101907
TA貢獻(xiàn)1887條經(jīng)驗(yàn) 獲得超5個贊
np.linalg.det(a)似乎工作得很好，并且運(yùn)行時間明顯更好：
a = np.random.rand(100,3,3)
%timeit -n 100 [np.linalg.det(e) for e in a]
626 μs ± 26.9 μs per loop (mean ± std. dev. of 7 runs, 100 loops each)
%timeit -n 100 np.linalg.det(a)
33.9 μs ± 7.08 μs per loop (mean ± std. dev. of 7 runs, 100 loops each)
添加回答
舉報(bào)
0/150
提交
取消