(self, input_resolution, dim, norm_layer=nn.LayerNorm)
| 305 | """ |
| 306 | |
| 307 | def __init__(self, input_resolution, dim, norm_layer=nn.LayerNorm): |
| 308 | super().__init__() |
| 309 | self.input_resolution = input_resolution |
| 310 | self.dim = dim |
| 311 | self.reduction = nn.Linear(4 * dim, 2 * dim, bias=False) |
| 312 | self.norm = norm_layer(4 * dim) |
| 313 | |
| 314 | def forward(self, x): |
| 315 | """ |