Скажем, у меня есть два массива 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
на самом деле довольно большие.