Умножение трехмерных матриц без зацикливания в python

Я хочу выполнить следующую операцию в python (numpy).

Matrix A is M x N x R
Matrix B is N x 1 x R

Умножение матрицы AB = C, где C - матрица M x 1 x R. По сути, каждый слой M x N в A (R из них) представляет собой матрицу, независимо умноженную на каждый вектор N x 1 в B. Я уверен, что это однострочный. Я пытался использовать tensordot (), но мне кажется, что это дает мне ответы, которых я не ожидаю.

Я программировал на Igor Pro почти 10 лет, и сейчас я пытаюсь преобразовать его страницы в Python.