(self, embed_ind)
| 296 | return embed_in |
| 297 | |
| 298 | def decode(self, embed_ind): |
| 299 | quantize = self._codebook.decode(embed_ind) |
| 300 | quantize = self.project_out(quantize) |
| 301 | quantize = rearrange(quantize, "b n d -> b d n") |
| 302 | return quantize |
| 303 | |
| 304 | def forward(self, x): |
| 305 | device = x.device |