[aiyagari] MAINT: transfer np.sum(a * b) to a @ b

[aiyagari] MAINT: transfer np.sum(a * b) to a @ b #124