w1 = torch.randn(784, 50) * math.sqrt(2/784) b1 = torch.randn(50) ... from torch.nn import init# linear layer implementation
確定! 回上一頁