Module attention

Module attention 

Source
Expand description

Attention Based Building Blocks

Structsยง

AttentionBlock
AttentionBlockConfig
Configuration for an attention block.
CrossAttention
SpatialTransformer
SpatialTransformerConfig