Fix requires_grad=True needs float tensor

timctho · timctho · commit 2bfebb0a4a83 · 2018-08-05T23:59:17.000+08:00
diff --git a/tutorials/01-basics/pytorch_basics/main.py b/tutorials/01-basics/pytorch_basics/main.py
@@ -23,9 +23,9 @@
 # ================================================================== #
 
 # Create tensors.
-x = torch.tensor(1, requires_grad=True)
-w = torch.tensor(2, requires_grad=True)
-b = torch.tensor(3, requires_grad=True)
+x = torch.tensor(1., requires_grad=True)
+w = torch.tensor(2., requires_grad=True)
+b = torch.tensor(3., requires_grad=True)
 
 # Build a computational graph.
 y = w * x + b    # y = 2 * x + 3
