Codebook laten dim visualization code?

[willxxy]

Hello,

Thank you for the great work. I was wondering if the code for visualizing the latent dim like the attached picture is available.