fix decomposition of aten.diag_embed (#120549)
Fixes #117019
Make the input that one dim negative and the other nonnegative be correctly solved in decomposition of `aten.diag_embed`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/120549
Approved by: https://github.com/Dalian991, https://github.com/janeyx99