human
I am working on the file "DL0110EN/2.6.2_multiple_linear_regression_training.ipynb".
The first task is:
<a id="ref1"></a>
<h2 align=center>Create the Model, Optimizer, and Total Loss Function (cost)</h2>

Create a custom module:
Can you write Python code for it?
assistant
#Here is the Python code you requested.

class linear_regression(nn.Module):
    def __init__(self,input_size,output_size):
        super(linear_regression,self).__init__()
        self.linear=nn.Linear(input_size,output_size)
    def forward(self,x):
        yhat=self.linear(x)
        return yhat