Enable diag for bool Tensors (#47455)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47455
Test Plan: Imported from OSS
Reviewed By: bdhirsh
Differential Revision: D24772483
Pulled By: H-Huang
fbshipit-source-id: 08ea4af4352972617db3c6475943b326f36b3049
