Method forward

beginner_source/onnx/onnx_registry_tutorial.py:139–140

(self, input_x)

class GeluModel(torch.nn.Module):
    def forward(self, input_x):
        return torch.ops.aten.gelu(input_x)


# Create a namespace for the custom operator using ONNX Script