about one_hot

[Sky-zhan]

trafficstars

Thanks for sharing your helpful work. There is a question I am wondering, that is the code ‘dist = (...)’ and ‘embed_onehot = F.one_hot(...)’, existing in class Quantize. I didn't find the corresponding content in the original paper，i.e., ‘Generating Diverse High-Fidelity Images with VQ-VAE-2’. Could you tell me where it comes from? Thanks a lot.

[rosinality]

You can refer to VQ-VAE implementation of Sonnet. (https://github.com/deepmind/sonnet/blob/master/sonnet/python/modules/nets/vqvae.py) It is for efficiently implement quantization.

[Sky-zhan]

thanks