Method forward

beginner_source/onnx/onnx_registry_tutorial.py:214–215

(self, input)

class AddAndRoundModel(torch.nn.Module):
    def forward(self, input):
        return add_and_round_op(input)


# Implement the custom operator in ONNX using ONNX Script