input: def create_linear_regression_model(input_dim: int, output_dim: int) -> LinearRegressionModel:
      """Creates a neural network class in Python to model a linear regression function.
      >>> model = create_linear_regression_model(2, 1)
      >>> model.forward(np.array([[1, 2]]))
      array([[0.]])
      """
output:     class LinearRegressionModel(nn.Module):
          def __init__(self, input_dim, output_dim):
              super(LinearRegressionModel, self).__init__()
              self.linear = nn.Linear(input_dim, output_dim)
  
          def forward(self, x):
              return self.linear(x)
  
      return LinearRegressionModel(input_dim, output_dim)