Для петлевого эквивалента для выражения einsum - PullRequest
Новая
       50

Для петлевого эквивалента для выражения einsum

1 голос
/ 20 сентября 2019

У меня есть следующее выражение суммы Эйнштейна (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, я следовал этому учебнику , который показывает, как реализовать некоторые выражения einsum с помощью вложенных циклов for, но это не сильно помогло.

Кто-нибудь знает, как это сделать?

1 Ответ

1 голос
/ 20 сентября 2019

Благодаря @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
...