How to get figure 2

[lm1687019806]

Hi authors:
I have read this excellent paper recently. Thank you so much for your paper. It really helped me to understand the differences between W space and Wk space. I want to know how to get a 2D visualization example of W and Wk space like figure 2.
Could you kindly help me resolve this question in your spare time?

[Booooooooooo]

I am also curious about this. Please kindly let my know if you figure out how. Thank you in advance.