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

fix-develop-build.sh
Yi Wang 7 years ago committed by GitHub
parent 6de0a18d5f
commit 227d8066e1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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()
```

Loading…
Cancel
Save