等效于 einsum 表达式的 For 循环
For loop equivalent for einsum expression
我有以下爱因斯坦求和 (einsum) 表达式,
import numpy as np
x = np.random.rand(1,8,2,8,10)
y = np.random.rand(8,10,10)
z = np.einsum('nkctv,kvw->nctw', x, y)
print (z.shape)
输出为:
(1, 2, 8, 10)
我想做的是使用一组嵌套的 for 循环构建等效的功能,我遵循了这个 tutorial 它展示了如何使用嵌套的 for 循环实现一些 einsum 表达式,但是它没有那么大帮助。
有人知道怎么做吗?
感谢@hpaulj,
解决方法是:
R = np.zeros((1,2,8,13))
for n in range(1):
for c in range(2):
for t in range(8):
for w in range(13):
total = 0
# These are the variables to sum over
for v in range(13):
for k in range(8):
total += x[n,k,c,t,v]*y[k,v,w]
R[n,c,t,w] = total
我有以下爱因斯坦求和 (einsum) 表达式,
import numpy as np
x = np.random.rand(1,8,2,8,10)
y = np.random.rand(8,10,10)
z = np.einsum('nkctv,kvw->nctw', x, y)
print (z.shape)
输出为:
(1, 2, 8, 10)
我想做的是使用一组嵌套的 for 循环构建等效的功能,我遵循了这个 tutorial 它展示了如何使用嵌套的 for 循环实现一些 einsum 表达式,但是它没有那么大帮助。
有人知道怎么做吗?
感谢@hpaulj,
解决方法是:
R = np.zeros((1,2,8,13))
for n in range(1):
for c in range(2):
for t in range(8):
for w in range(13):
total = 0
# These are the variables to sum over
for v in range(13):
for k in range(8):
total += x[n,k,c,t,v]*y[k,v,w]
R[n,c,t,w] = total