Class Module

test/export/test_serialize.py:576–582

        linear = torch.nn.Linear(2, 2)

        class Module(torch.nn.Module):
            def forward(self, x):
                x = x + 1
                y = x.t()
                y = y.relu()
                y = linear(y)
                return y

        ep = export(Module(), inp)