neodroidvision.classification.mechanims.attention.self.self_attention

Functions

init_weights(m)

param m

Classes

SelfAttentionModule(in_channels)

Self attention Layer