W = np.zeros((D, H))
W = np.random.rand(D, H)
W = 0.01 * np.random.randn(D, H)
W = np.ones((D, H))
torch.manual_seed(seed)