Method decode

funasr/utils/kws_utils.py:295–308

Get an initial state for decoding. Args: x (torch.Tensor): The encoded feature tensor Returns: decode result

(self, x: torch.Tensor)

    def decode(self, x: torch.Tensor):
        """Get an initial state for decoding.

        Args:
            x (torch.Tensor): The encoded feature tensor

        Returns: decode result

        """

        raw_logp = self.ctc.softmax(x.unsqueeze(0)).detach().squeeze(0).cpu()
        xlen = torch.tensor([raw_logp.size(1)])

        return self._decode_inside(raw_logp, xlen)