def forward()

in opacus_lab/models/GPT2/model/attention.py [0:0]

• 7 lines of code
• 1 McCabe index (conditional complexity)

    def forward(
        self,
        q: torch.Tensor,
        k: torch.Tensor,
        v: torch.Tensor,
        past: Optional[Past] = None,
        mask: Optional[torch.Tensor] = None,