* add diag_embed op (#23385)

* add diag_embed op, test=release/2.0-beta

* solved a conflict, test=release/2.0-beta