Copied from torch
's linear algebra test file.
torch.einsum('i->', x) # sum
torch.einsum('i,i->', x, x) # dot
torch.einsum('i,i->i', x, x) # vector element-wisem mul
torch.einsum('i,j->ij', x, y) # outer