未验证 提交 227d8066 编写于 作者: Y Yi Wang 提交者: GitHub

Fix a small bug in the example code snippet (#13286)

上级 6de0a18d
......@@ -30,7 +30,7 @@ x = Variable(randn(20, 1)))
label = Variable(randint(1))
W_1, W_2 = Variable(randn(20, 20)), Variable(randn(10, 20))
h = matmul(W_1, x)
pred = matmul(W_2, x)
pred = matmul(W_2, h)
loss = softmax(pred, label)
loss.backward()
```
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册