Skip to content

Commit 50a4b46

Browse files
committed
PyTorch now requires float Tensor for gradient computation
1 parent 659a4ae commit 50a4b46

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

03-autograd_tutorial.ipynb

+4-4
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,7 @@
221221
"metadata": {},
222222
"outputs": [],
223223
"source": [
224-
"x = torch.arange(1, n + 1, requires_grad=True)\n",
224+
"x = torch.arange(1., n + 1, requires_grad=True)\n",
225225
"w = torch.ones(n, requires_grad=True)\n",
226226
"z = w @ x\n",
227227
"z.backward()\n",
@@ -234,7 +234,7 @@
234234
"metadata": {},
235235
"outputs": [],
236236
"source": [
237-
"x = torch.arange(1, n + 1)\n",
237+
"x = torch.arange(1., n + 1)\n",
238238
"w = torch.ones(n, requires_grad=True)\n",
239239
"z = w @ x\n",
240240
"z.backward()\n",
@@ -248,7 +248,7 @@
248248
"outputs": [],
249249
"source": [
250250
"with torch.no_grad():\n",
251-
" x = torch.arange(1, n + 1)\n",
251+
" x = torch.arange(1., n + 1)\n",
252252
" w = torch.ones(n, requires_grad=True)\n",
253253
" z = w @ x\n",
254254
" z.backward()\n",
@@ -283,7 +283,7 @@
283283
"name": "python",
284284
"nbconvert_exporter": "python",
285285
"pygments_lexer": "ipython3",
286-
"version": "3.6.5"
286+
"version": "3.6.6"
287287
}
288288
},
289289
"nbformat": 4,

0 commit comments

Comments
 (0)