Method forward

tests/testing_data/bundle_test_network.py:25–29

(self, x: torch.Tensor, extra_arg1: int, extra_arg2: int)

    """

    def forward(self, x: torch.Tensor, extra_arg1: int, extra_arg2: int) -> torch.Tensor:  # type: ignore
        x = self.model(x)
        x += extra_arg1
        x += extra_arg2
        return x