I'm studying the Codebook Matching architecture and noticed what appears to be a discrepancy between the paper description and the code implementation regarding the Encoder's input. I would greatly ...