allow for access to codebook with property method

Author: lucidrains

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'vector_quantize_pytorch',
   packages = find_packages(),
-  version = '0.2.1',
+  version = '0.2.2',
   license='MIT',
   description = 'Vector Quantization - Pytorch',
   author = 'Phil Wang',

vector_quantize_pytorch/vector_quantize_pytorch.py

@@ -38,6 +38,10 @@ def __init__(
         self.register_buffer('cluster_size', torch.zeros(n_embed))
         self.register_buffer('embed_avg', embed.clone())
 
+    @property
+    def codebook(self):
+        return self.embed.transpose(0, 1)
+
     def forward(self, input):
         dtype = input.dtype
         flatten = input.reshape(-1, self.dim)