python – 使用多个内核的Numpy np.einsum数组乘法
发布时间:2020-08-05 07:08:49 所属栏目:Python 来源:互联网
导读:我已经编译了numpy 1.6.2和scipy与MKL希望有更好的性能.目前我有一个严重依赖于np.einsum()的代码,我被告知einsum对MKL不好,因为几乎没有矢量化. =(所以我想用np.dot()和切片来重写我的一些代码,只是为了能够获得一些多核速度.我非常喜欢np.einsum()的简单性,
我已经编译了numpy 1.6.2和scipy与MKL希望有更好的性能.
那么如何在np.dot()高效的MKL操作中转换这样的东西,或者其他3,4和5维数组乘法呢? 我会提供更多信息: 为此,我使用的代码:
这不是那么快,在cython中编码的同样的东西快5倍:
有没有办法在纯Python中用cython的性能来做到这一点? (我还没弄清楚如何对这个方程进行推理)
它也将使用多个核心,如numpy.dot(). (编辑:阜阳站长网) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |
推荐文章
站长推荐
热点阅读