    def forward(self, x):
        """
        In the forward function we accept a Tensor of input data
        """
        h_relu = self.linear1(x).clamp(min=0)
        y_pred = self.linear2(h_relu)
        return y_pred
In the forward function we accept a Tensor of input data
tests/integration/frameworks/models/onnx.py