def forward()

in models/vision_transformer.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

    def forward(self, x):
        if self.normalize_inputs is not None:
            if self.normalize_inputs == NORMALIZE_L2:
                x = nn.functional.normalize(x, p=2.0, dim=1)
        return self.layers(x)