    def forward(self, x, x_mask):
        """
        Args:
            x: (N, seq_len, input_dim)
        Returns:
            output: (N, seq_len, output_dim)
        """
        # (N, seq_len, d_model)
        x = x.transpose(1, 2)
        x_mask = x_mask.transpose(1, 2)
        output = self.conformer_encoder(x, x_mask)
        # (N, seq_len, output_dim)
        output = self.projection(output)
        output = output.transpose(1, 2)
        return output
Args:
x: (N, seq_len, input_dim)
Returns:
output: (N, seq_len, output_dim)
models/svc/transformer/conformer.py