Method forward

Image_Classification/src/models/swin.py:14–15

(self, x)

        self.displacement = displacement

    def forward(self, x):
        return torch.roll(x, shifts=(self.displacement, self.displacement), dims=(1, 2))


class Residual(nn.Module):