multiplicación de vector de matriz numpy

| | | | | | | | | |

Cuando multiplico dos matrices numpy de tamaños (nxn)*(nx 1), obtengo una matriz de tamaño (nxn). Siguiendo las reglas normales de multiplicación de matrices, se espera un vector (nx 1), pero simplemente no puedo encontrar ninguna información sobre cómo se hace esto en el módulo Numpy de Python.

Lo que pasa es que no quiero implementarlo manualmente para preservar la velocidad del programa.

El código de ejemplo se muestra a continuación:

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

Lo que quiero es:

print a*b >> ; [16 6 8]