virne.solver.learning.neural_network.att#

Classes

MultiHeadAttentionWithOutput(embedding_dim)

MultiHeadSelfAttention(input_dim[, ...])

PositionalEncoder(embedding_dim[, ...])