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]])
reply
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