Method forward

intermediate_source/torch_export_tutorial.py:969–971

(self, x)

        self.linear = torch.nn.Linear(3, 3)

    def forward(self, x):
        x = self.linear(x)
        return x

inp = torch.randn(2, 3, device="cuda")
m = M().to(device="cuda")