标签： intel-mkl

python – 使用多个内核的Numpy np.einsum数组乘法

dawei Python 2020年8月4日

我已经编译了numpy 1.6.2和scipy与MKL希望有更好的性能.目前我有一个严重依赖于np.einsum()的代码,我被告知einsum对MKL不好,因为几乎没有矢量化. =(所以我想用np.dot()和切片来重写我的一些代码,只是为了能够获得一些多核速度.我非常喜欢np.einsum()的简单性,

返回顶部