miniworld.models_MiniWorld_v1_5_use_interaction.Attention_module

Classes

Attention(d_query, d_key, n_head, d_hidden, ...)
AttentionWithBias([d_in, d_bias, n_head, ...])
BiasedAxialAttention(d_pair, d_bias, n_head, ...)
FeedForwardLayer(d_model, r_ff[, p_drop])
GraphTriangleAttention(d_pair, d_state[, ...])
MSAColAttention([d_msa, n_head, d_hidden])
MSAColGlobalAttention([d_msa, n_head, d_hidden])
MSARowAttentionWithBias([d_msa, d_pair, ...])
SequenceWeight(d_msa, n_head, d_hidden[, p_drop])
StateAttentionGate(d_state, d_pair[, ...])
TriangleMultiplication(d_pair[, d_hidden, ...])