Функция умножения матричной 3d-матрицы

Question

Предположим, что у меня есть ndarray, W формы (m, n, n) и вектор C размерности (m, n). Мне нужно умножить эти два следующим образом

result = np.empty(m,n) 
for i in range(m): 
    result[i] = W[i] @ C[i] 

Как это сделать в векторизованном пути без петель и все?

Answer 1

С, вам нужно держать первую ось от обоих W и C выровненный, в то время как потери последней оси из них с матрицей-умножения, я бы предлагают использовать np.einsum для очень эффективного подхода, как так -

np.einsum('ijk,ik->ij',W,C)

np.tensordot или np.dot не имеет функции для выравнивания осей, и здесь улучшается np.einsum.

Answer 2

np.dot должен быть вашим другом. http://docs.scipy.org/doc/numpy-1.10.0/reference/generated/numpy.dot.html

Answer 3

Это делается с помощью np.tensordot

ans=np.tensordot(W,C,axes=[2,1])[np.arange(m),:,np.arange(m)] 
assert np.all(result==ans)