From 9493a8e83ac0e31aba9a06cf813da1e9de8f7efe Mon Sep 17 00:00:00 2001 From: Gajendra Jung Katuwal Date: Tue, 6 Mar 2018 16:24:00 -0500 Subject: [PATCH] Passing index as variable in index_select() --- Capsule Network.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Capsule Network.ipynb b/Capsule Network.ipynb index f3cd682..abe879a 100644 --- a/Capsule Network.ipynb +++ b/Capsule Network.ipynb @@ -170,7 +170,7 @@ " masked = Variable(torch.sparse.torch.eye(10))\n", " if USE_CUDA:\n", " masked = masked.cuda()\n", - " masked = masked.index_select(dim=0, index=max_length_indices.squeeze(1).data)\n", + " masked = masked.index_select(dim=0, index=max_length_indices.squeeze(1))\n", " \n", " reconstructions = self.reconstraction_layers((x * masked[:, :, None, None]).view(x.size(0), -1))\n", " reconstructions = reconstructions.view(-1, 1, 28, 28)\n",