OpInfo: `diag_embed`, `diagonal` (#58642)
Summary:
See: https://github.com/pytorch/pytorch/issues/54261.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58642
Reviewed By: ngimel
Differential Revision: D28627226
Pulled By: mruberry
fbshipit-source-id: b96fa8410bd53937ddb72a46c02b949691ee9458