Как расставить подрешетки в NumPy?

Question

Я хочу выполнить точечное произведение между подмассивами двух массивов.
Вот код, использующий цикл.
Есть ли более лаконичный способ выполнить такую ​​операцию без циклов?

result = numpy.empty(100)
A = numpy.random.randn(100, 1, 3)
B = numpy.random.randn(100, 3, 1)
for i in range(100):
    result[i] = A[i].dot(B[i])

Answer 1

U сделал лучший ответ. Я использую:

C = numpy.transpose(numpy.sum(A*numpy.reshape(B,(100,1,3)),2))

Answer 2

Я обнаружил, что numpy.einsum() отлично бы удовлетворил мою потребность.

numpy.einsum('ijk,ikj->i', A, B)