pytorch

45b33c83 - Revert "Revert D24923679: Fixed einsum compatibility/performance issues (#46398)" (#49189)

Revert "Revert D24923679: Fixed einsum compatibility/performance issues (#46398)" (#49189) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/49189 This reverts commit d307601365c3b848072b8b8381208aedc1a0aca5 and fixes the bug with diagonals and ellipsis combined. Test Plan: Imported from OSS Reviewed By: glaringlee Differential Revision: D25540722 Pulled By: heitorschueroff fbshipit-source-id: 86d0c9a7dcfda600b546457dad102af2ff33e353