Я пытаюсь умножить два трехмерных массива форм M1 (10, 3, 4) и M2 (25, 4, 1). Я могу выполнить операцию как

В настоящее время я выполняю эту операцию с помощью цикла.

for i in range(M2.shape[0]):
        a[i] = np.matmul(M1,M2[i].reshape(1,4,1))

И я добавляю результат из каждого цикла и создаю матрицу. Я хочу избежать цикла for и попытаться выполнить ту же операцию, только используя матричную операцию. Есть ли способ выполнить эту операцию, используя numpy или любую другую библиотеку в python? Пожалуйста, предоставьте образец кода для этого.

-1
Rishi 12 Апр 2019 в 08:56

2 ответа

Лучший ответ

Вы можете использовать np.einsum, чтобы решить это

M1 = np.random.randn(10,3,4)
M2 = np.random.randn(25,4,1)
out = np.einsum('abc,dce->dabe', M1, M2)
print(out.shape)
Out: (25, 10, 3, 1)

Выше abc - это размеры M1, а dce - это размеры M2. Обратите внимание, c является общим измерением в M1 и M2.

0
Atul Shanbhag 12 Апр 2019 в 06:10

Попробуйте numpy.tensordot, что и используется для многомерных матричных умножений

import numpy as np

a = np.arange(60.).reshape(3,4,5)
b = np.arange(24.).reshape(4,3,2)
c = np.tensordot(a,b, axes=([1,0],[0,1]))
0
Muthukrishnan 12 Апр 2019 в 06:04