Method __init__

monai/networks/layers/vector_quantizer.py:207–212

(self, quantizer: EMAQuantizer)

    """

    def __init__(self, quantizer: EMAQuantizer):
        super().__init__()

        self.quantizer: EMAQuantizer = quantizer

        self.perplexity: torch.Tensor = torch.rand(1)

    def forward(self, inputs: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor]:
        quantized, loss, encoding_indices = self.quantizer(inputs)