VectorQuantizer

The VQ-VAE sequence vector quantizer.

Abstract Signature:

VectorQuantizer(num_latent_classes: int, latent_dim: int, beta: float)

PyTorchPaxML / Praxis

PyTorch

API: —

Strategy: Custom / Partial

PaxML / Praxis

API: praxis.layers.VectorQuantizer

Strategy: Direct Mapping

Official Docs ↗