code: def __init__(self):
      super().__init__()
      self.weights = nn.Parameter(
          torch.randn(784, 10) / math.sqrt(784)
      )  # W
      self.bias = nn.Parameter(torch.zeros(10))
  
imports: from torch import nn
label_desc: Use image node_id 1 to create a new MnistLogistic object from inherited base classes: nn with example: obj = MnistLogistic()