Fix incorrect __torch_function__ handling in einsum (#38741)
Summary:
Closes gh-38479
Pull Request resolved: https://github.com/pytorch/pytorch/pull/38741
Differential Revision: D21662512
Pulled By: ezyang
fbshipit-source-id: 247e3b50b8f2dd842c03be8d6ebe71910b619bc6
