avoiding for-loop 

i'll try to figure out how to avoid ``tmp`` calculation with ``np.einsum`` if it possible:)
Merge pull request #3 from ipashchenko/patch-1

avoiding for-loop