Args: x: Tensor input tensor [batch, length, features] mask: Tensor mask tensor [batch, length] factor: int squeeze factor (default 2) Returns: Tensor1, Tensor2 squeezed x [batch, length // factor, factor * features]
(x: torch.Tensor, mask: torch.Tensor, factor: int = 2)