diff --git a/README.md b/README.md index 6e479f2342..291416c3b4 100644 --- a/README.md +++ b/README.md @@ -44,8 +44,8 @@ y = Tensor([[2.0,0,-2.0]], requires_grad=True) z = y.matmul(x).sum() z.backward() -print(x.grad) # dz/dx -print(y.grad) # dz/dy +print(x.grad.numpy()) # dz/dx +print(y.grad.numpy()) # dz/dy ``` ### Same example in torch