src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py/FlaxWav2Vec2PositionalConvEmbedding/__call__
class FlaxWav2Vec2PositionalConvEmbedding:    def __call__(self, hidden_states):
        hidden_states = hidden_states.transpose((0, 1, 2))

        hidden_states = self.conv(hidden_states)

        if self.num_pad_remove > 0:
            hidden_states = hidden_states[:, : -self.num_pad_remove, :]
        hidden_states = self.activation(hidden_states)

        hidden_states = hidden_states.transpose((0, 1, 2))
        return hidden_states