human
I am working on the file "DL0110EN/3.3.2lab_predicting _MNIST_using_Softmax.ipynb".
The first task is:
<a id="ref3"></a>
Build a Softmax Classifer
Build a Softmax classifier class:
Can you write Python code for it?
 Write me a the Python code you requested?
assistant
#Here is the Python code you requested.

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