Method __init__

method/nn_util.py:348–350

(self, embed_dim=96)

    """

    def __init__(self, embed_dim=96):
        super().__init__()
        self.embed_dim = embed_dim

    def forward(self, x, x_size):
        B, H, W, C = x.shape