mmlearn.modules.layers package

Custom, reusable layers for models and tasks.

Submodules

• mmlearn.modules.layers.attention module
  • Attention
    • Attention.forward()
• mmlearn.modules.layers.embedding module
  • ConvEmbed
    • ConvEmbed.forward()
  • PatchEmbed
    • PatchEmbed.forward()
  • get_1d_sincos_pos_embed()
  • get_1d_sincos_pos_embed_from_grid()
  • get_2d_sincos_pos_embed()
  • get_2d_sincos_pos_embed_from_grid()
• mmlearn.modules.layers.logit_scaling module
  • LearnableLogitScaling
    • LearnableLogitScaling.extra_repr()
    • LearnableLogitScaling.forward()
• mmlearn.modules.layers.mlp module
  • MLP
• mmlearn.modules.layers.normalization module
  • L2Norm
    • L2Norm.forward()
• mmlearn.modules.layers.patch_dropout module
  • PatchDropout
    • PatchDropout.forward()
    • PatchDropout.uniform_mask()
• mmlearn.modules.layers.transformer_block module
  • Block
    • Block.forward()
  • DropPath
    • DropPath.forward()
  • drop_path()