Умножение трехмерных матриц в Numpy - PullRequest
0 голосов
/ 04 ноября 2018

Мне нужно умножить две двумерные матрицы, bob и tim, в Numpy Python 3.x

bob.shape дает (2,4)

tim.shape дает (7,4)

Этот фрагмент кода дает трехмерную матрицу с формой (2,7,4)

np.array([foo*tim for foo in bob])

Это дает вывод, который я хочу. Но мне было интересно, есть ли более элегантный / более быстрый способ сделать это в NumPy, чем мне нужно перебирать bob

1 Ответ

0 голосов
/ 04 ноября 2018

См. Python Broadcasting

bob.reshape((2, 1, 4)) * tim.reshape((1, 7, 4))
...