↓ 1 callersMethod__init__(self, in_channels, out_channels, n_heads, seq_len, modes=0, mode_select_method='random')
layers/FourierCorrelation.py:29
↓ 1 callersMethod__init__(self, mask_flag=True, factor=1, scale=None, attention_dropout=0.1, output_attention=False)
layers/AutoCorrelation.py:19
↓ 1 callersMethod_update_context(self, context_in, V, scores, index, L_Q, attn_mask)
layers/SelfAttention_Family.py:126