from torch import nn

class Mnist_Logistic([nn.Module](https://docs.pytorch.org/docs/stable/generated/torch.nn.Module.html#torch.nn.Module "torch.nn.Module")):
    def __init__(self):
        super().__init__()
        self.[weights](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor") = [nn.Parameter](https://docs.pytorch.org/docs/stable/generated/torch.nn.parameter.Parameter.html#torch.nn.parameter.Parameter "torch.nn.parameter.Parameter")([torch.randn](https://docs.pytorch.org/docs/stable/generated/torch.randn.html#torch.randn "torch.randn")(784, 10) / math.sqrt(784))
        self.[bias](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor") = [nn.Parameter](https://docs.pytorch.org/docs/stable/generated/torch.nn.parameter.Parameter.html#torch.nn.parameter.Parameter "torch.nn.parameter.Parameter")([torch.zeros](https://docs.pytorch.org/docs/stable/generated/torch.zeros.html#torch.zeros "torch.zeros")(10))

    def forward(self, [xb](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor")):
        return [xb](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor") @ self.[weights](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor") + self.[bias](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor")