def forward()

in ViT4MNIST/vit_pytorch.py [0:0]

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

    def forward(self, x, mask=None):
        for attn, ff in self.layers:
            x = attn(x, mask=mask)
            x = ff(x)
        return x