I thought so too, but it doesn't seem to work since X has shape (3,).

This seems to work,

   Z = Y @ X[:,np.newaxis]