Support plot norm of specific embeddings of a LUT in diagnose_options (#19809)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19809
as title
Reviewed By: chocjy
Differential Revision: D15100505
fbshipit-source-id: cba290fd4317b260e2bf1689b9ca215d3d19a9e2