import numpy as np
 
# (3,) · (3, 2) -> (2,)
vec = np.array([1.0, 1.0, 1.0])
mat = np.array([[1.0, 2.0],
                [3.0, 4.0],
                [5.0, 6.0]])
 
vec_mat_result = np.dot(vec, mat)
print(vec_mat_result.shape)
print(vec_mat_result)
 
# (2, 3) · (3,) -> (2,)
matrix = np.array([[1.0, 2.0, 3.0],
                   [4.0, 5.0, 6.0]])
vector = np.array([1.0, 1.0, 1.0])
 
mat_vec_result = np.dot(matrix, vector)
print(mat_vec_result.shape)
print(mat_vec_result)
 

aW1wb3J0IG51bXB5IGFzIG5wCgojICgzLCkgwrcgKDMsIDIpIC0+ICgyLCkKdmVjID0gbnAuYXJyYXkoWzEuMCwgMS4wLCAxLjBdKQptYXQgPSBucC5hcnJheShbWzEuMCwgMi4wXSwKICAgICAgICAgICAgICAgIFszLjAsIDQuMF0sCiAgICAgICAgICAgICAgICBbNS4wLCA2LjBdXSkKCnZlY19tYXRfcmVzdWx0ID0gbnAuZG90KHZlYywgbWF0KQpwcmludCh2ZWNfbWF0X3Jlc3VsdC5zaGFwZSkKcHJpbnQodmVjX21hdF9yZXN1bHQpCgojICgyLCAzKSDCtyAoMywpIC0+ICgyLCkKbWF0cml4ID0gbnAuYXJyYXkoW1sxLjAsIDIuMCwgMy4wXSwKICAgICAgICAgICAgICAgICAgIFs0LjAsIDUuMCwgNi4wXV0pCnZlY3RvciA9IG5wLmFycmF5KFsxLjAsIDEuMCwgMS4wXSkKCm1hdF92ZWNfcmVzdWx0ID0gbnAuZG90KG1hdHJpeCwgdmVjdG9yKQpwcmludChtYXRfdmVjX3Jlc3VsdC5zaGFwZSkKcHJpbnQobWF0X3ZlY19yZXN1bHQpCg==