When I try
class Feedforward(nn.Module):
    def __init__(self):
        super(Feedforward, self).__init__()
        prob_drop = 0.0
        self.fc1 = nn.init.normal(nn.Linear(784, 200))

I get
File "model/network.py", line 20, in __init__
    self.fc1 = torch.nn.init.normal(nn.Linear(784, 200))
AttributeError: 'module' object has no attribute 'init'