小编sem*_*mmo的帖子

当输入是许多相同的数组时,使 np.einsum 更快?(或任何其他更快的方法)

我有一段代码类型:

nnt = np.real(np.einsum('xa,xb,yc,yd,abcde->exy',evec,evec,evec,evec,quartic))
Run Code Online (Sandbox Code Playgroud)

其中evec是(比如说)一个 L x Lnp.float32阵列,并且quartic是一个 L x L x L x L x Tnp.complex64阵列。

我发现这个例程相当慢。

我认为由于所有evec的都是相同的,可能有更快的方法吗?

提前致谢。

python numpy numpy-einsum

6
推荐指数
1
解决办法
64
查看次数

标签 统计

numpy ×1

numpy-einsum ×1

python ×1