D Question 1 1 class SoftMaxRegression(nn. Module) : def _init_(self): super()._init_() # define layers self. flatten = nn. Flatten() self. linear = nn. Linear(784, 10) 8 def forward(self, x): # Now