Поэлементное умножение массивов разных форм в python

Скажем, у меня есть два массива aи b,

  a.shape = (5,2,3)
  b.shape = (2,3)

, тогда c = a * bдаст мне массив cформы (5,2,3)с c[i,j,k] = a[i, j,k]*b[j,k],

Теперь ситуация такова,

  a.shape = (5,2,3)
  b.shape = (2,3,8)

и я хочу, чтобы cимел форму (5,2,3, 8)с c[i,j,k,l] = a[i,j,k]*b[j,k,l]

Как это сделать эффективно? Мои aи bна самом деле довольно большие.

9
задан user1342516 20 May 2012 в 21:53
поделиться