Replace np.sum(a * b) with a @ b for better performance and accuracy

Replace np.sum(a * b) with a @ b for better performance and accuracy #249