elektronn3.modules.axial_attention module¶
- class elektronn3.modules.axial_attention.AxialAttention(*args: Any, **kwargs: Any)[source]¶
Bases:
Module
- class elektronn3.modules.axial_attention.AxialImageTransformer(*args: Any, **kwargs: Any)[source]¶
Bases:
Module
- class elektronn3.modules.axial_attention.AxialPositionalEmbedding(*args: Any, **kwargs: Any)[source]¶
Bases:
Module
- class elektronn3.modules.axial_attention.Deterministic(*args: Any, **kwargs: Any)[source]¶
Bases:
Module
- class elektronn3.modules.axial_attention.IrreversibleBlock(*args: Any, **kwargs: Any)[source]¶
Bases:
Module
- class elektronn3.modules.axial_attention.PermuteToFrom(*args: Any, **kwargs: Any)[source]¶
Bases:
Module
- class elektronn3.modules.axial_attention.ReversibleBlock(*args: Any, **kwargs: Any)[source]¶
Bases:
Module
- class elektronn3.modules.axial_attention.ReversibleSequence(*args: Any, **kwargs: Any)[source]¶
Bases:
Module
- class elektronn3.modules.axial_attention.SelfAttention(*args: Any, **kwargs: Any)[source]¶
Bases:
Module