Method __init__

method/nn_util.py:329–331

(self, embed_dim=96, norm_layer=None)

    """

    def __init__(self, embed_dim=96, norm_layer=None):
        super().__init__()
        self.norm = nn.LayerNorm(embed_dim) if norm_layer is not None else None

    def forward(self, x):
        x = x.permute(0, 2, 3, 1)  # (b c h w) -> (b h w c)