Method quantize

monai/networks/layers/vector_quantizer.py:230–233

(self, encodings: torch.Tensor)

        return self.quantizer.embed(embedding_indices=embedding_indices)

    def quantize(self, encodings: torch.Tensor) -> torch.Tensor:
        output = self.quantizer(encodings)
        encoding_indices: torch.Tensor = output[2]
        return encoding_indices