    def cache_to_batch(self, cache: List[List[torch.Tensor]]) -> List[torch.Tensor]:
        """
        Args:
            cache : [[history_1], [history_2], [history3]...]

        Returns:
            new_caceh: [history],
                history: [bs, ...]
        """
        history = torch.cat([h[0] for h in cache], dim=0)
        return [history]
Args:
cache : [[history_1], [history_2], [history3]...]

Returns:
new_caceh: [history],
history: [bs, ...]
modules/wenet_extractor/transducer/predictor.py