numpy матричное векторное умножение

| | | | | | | | | |

Когда я умножаю два пустых массива размеров (nxn)*(nx 1), я получаю матрицу размера (nxn). Следуя обычным правилам умножения матриц, ожидается вектор (nx 1), но я просто не могу найти никакой информации о том, как это делается в модуле Python Numpy.

Дело в том, что я не хочу реализовать его вручную, чтобы сохранить скорость программы.

Пример кода показан ниже:

a = np.array([[ 5, 1 ,3], [ 1, 1 ,1], [ 1, 2 ,1]]) b = np.array([1, 2, 3]) print a*b >> [[5 2 9] [1 2 3] [1 4 3]] 

Я хочу:

напечатать a*b >> ; [16 6 8]