> To make `Z` a column vector, we would need something like `Z = (Y @ X)[:,np.newaxis]`.
Doesn't just (Y @ X)[None] work? None adding an extra dimension works in practice but I don't know if you're "supposed" to do that
Doesn't just (Y @ X)[None] work? None adding an extra dimension works in practice but I don't know if you're "supposed" to do that
(Y @ X)[None]
# array([[14, 32, 50]])
(Y @ X)[None].T
# array([[14],
# [32],
# [50]])