src/transformers/models/audio_spectrogram_transformer/modeling_audio_spectrogram_transformer.py/ASTAttention/__init__
class ASTAttention:    def __init__(self, config: ASTConfig) -> None:
        super().__init__()
        self.attention = ASTSelfAttention(config)
        self.output = ASTSelfOutput(config)
        self.pruned_heads = set()