我有一个关于einsum省略号的问题,我认为这个问题肯定会出现在StackExchange上,但不知怎么的,我似乎找不到。最好的方法是使它们成为向量,并修改我的einsum表达式,使它们能够接受任意数量的附加维度,这些维度只是简单地完成。这个问题是问这是否可能,如果是的话,如何做到。假设我有一个einsum表达式,它创建了某种矩阵乘法。', a, b)
当然,这会导致一个错误,但希望从示例中可以清楚地看到我
下面是一个简单的代码,它将4D矩阵a乘以3D矩阵b:import numpy as np return np.einsum('ijkl,jkl->ikl', a, b)
s0, s1, s2, s3%timeit original(a, b)
395 ms ± 2.2 ms per loop (mean ± std.